mindspore.ops.tensor_scatter_add

查看源文件
mindspore.ops.tensor_scatter_add(input_x, indices, updates)[源代码]

返回一个新tensor，根据指定索引和更新值对 input_x 进行加法更新。

\[output\left [indices \right ] = input\_x + update\]

下图是tensor_scatter_add计算过程的例子：

../../_images/TensorScatterAdd.png

说明

如果 indices 中的值超出输入 input_x 索引范围：

  • GPU平台上相应的 updates 不会更新到 input_x 且不会抛出索引错误。

  • CPU平台上直接抛出索引错误。

  • Ascend平台不支持越界检查，若越界可能会造成未知错误。

参数：

  • input_x (Tensor) - 输入tensor。 input_x 的维度必须不小于indices.shape[-1]。

  • indices (Tensor) - 输入tensor的索引，数据类型为mindspore.int32或mindspore.int64。其rank必须至少为2。

  • updates (Tensor) - 指定与 input_x 相加的Tensor，其数据类型与 input_x 相同，并且shape应等于 \(indices.shape[:-1] + input\_x.shape[indices.shape[-1]:]\) 。

返回：

Tensor

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> input_x = mindspore.tensor([[-0.1, 0.3, 3.6], [0.4, 0.5, -3.2]], mindspore.float32)
>>> indices = mindspore.tensor([[0, 0], [0, 0]], mindspore.int32)
>>> updates = mindspore.tensor([1.0, 2.2], mindspore.float32)
>>> output = mindspore.ops.tensor_scatter_add(input_x, indices, updates)
>>> print(output)
[[ 3.1  0.3  3.6]
 [ 0.4  0.5 -3.2]]