EmbeddingBag

功能

计算embedding函数的反向输出

边界

【输入】

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

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

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

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

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

【属性】

四个属性:

mode:string型

scale_grad_by_fraq:bool型

sparse:bool型

include_last_offset:bool型

【输出】

一个输出

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

支持的ONNX版本

自定义算子,无对应onnx版本