昇腾社区首页
中文
注册

ScatterND

功能

创建data的拷贝,同时在指定indices处根据updates更新。

输入

  • data:输入Tensor,数据类型支持float16、float。
  • indices:输入Tensor,indices.shape[-1]<=data维度。数据类型支持int64。
  • updates:输入Tensor,shape维度 = data维度+ indices维度 - indices_shape[-1] - 1,数据类型支持float16、float。

输出

y:输出Tensor,和输入data同样的数据类型和shape。

约束

indices的索引值不能为负数

支持的ONNX版本

Opset v11/v12/v13/v14/v15/v16/v17/v18