ScatterElements
功能
根据updates和indices来更新data的值,并把结果返回。与Scatter不同的是它提供了更精细的控制,允许指定要更新的元素在维度上的具体位置。
输入
- data: 输入Tensor,数据类型:float16、float、int32。
- indices:Tensor,数据类型:int32、int64。shape维度与输入data一致。
- updates:Tensor,数据类型:float16、float、int32。shape与indices维度一致。
输出
y:Tensor,和输入data的shape、数据类型一致。
属性
axis:int,默认是0,表示沿axis取数据。
支持的ONNX版本
Opset v11/v12/v13/v14/v15/v16/v17/v18
父主题: 支持ONNX算子清单