EmbeddingBag

功能

根据indices从weight中获得一组被聚合的数,然后根据offsets的偏移和mode指定的聚合模式对获取的数进行max、sum、mean聚合。其余参数则更细化了计算过程的控制。

输入

属性

输出

y:输出Tensor,数据类型:float。

约束与限制

无。

支持的ONNX版本

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