EmbeddingBag

功能

计算embedding函数的反向输出

边界

【输入】

两个必选输入,两个可选输入

weight:一个tensor,数据类型:float

indices:一个tensor,数据类型:int32

offset:一个tensor,数据类型:int32

per_sample_weights:一个tensor,数据类型:float

【属性】

四个属性:

mode:string型

scale_grad_by_fraq:bool型

sparse:bool型

include_last_offset:bool型

【输出】

一个输出

y:一个tensor,数据类型:float

支持的ONNX版本

Opset v8/v9/v10/v11/v12/v13/v14/v15/v16