EmbeddingBag

功能

计算embedding函数的反向输出。

输入

两个必选输入，两个可选输入：

weight：必选，一个tensor，数据类型：float。

indices：必选，一个tensor，数据类型：int32。

offset：可选，一个tensor，数据类型：int32。

per_sample_weights：可选，一个tensor，数据类型：float。

输出

一个输出：

y：一个tensor，数据类型：float。

属性

mode：string型。

scale_grad_by_fraq：bool型。

sparse：bool型。

include_last_offset：bool型。

支持的ONNX版本

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