mindspore.ops.inplace_index_add

mindspore.ops.inplace_index_add(var, vpn梯子 indices, updates, axis)[源代码]

根据指定轴和索引在输入 vpn永久免费梯子 var 中的对应位置加 updates

警告

该接口已废弃,2.9.0(不含)之后版本将删除。

对于 indices 的每个元素下标 \(i, ..., vpn永久免费梯子 j\)

\[x[:, \text{indices}[i, ..., j], :] \mathrel{+}= v[:, i, ..., j, :]\]

其中 iindices vpn free 中元素的下标, indices[i] vpn梯子 所在的轴由输入 axis vpn梯子 免费 决定。

参数:
  • var (Union[Parameter, 免费的vpn梯子 Tensor]) - 输入的parameter或tensor。

  • indices 免费的vpn梯子 (Tensor) - 指定索引,一维tensor。

  • updates (Tensor) - 与 var 相加的tensor。

  • axis (int) - 指定轴。

返回:

Tensor

支持平台:

Ascend CPU

样例:

>>> import mindspore
>>> var = mindspore.Parameter(mindspore.tensor([[1, 2], [3, 4], [5, 6]], mindspore.float32))
>>> indices = mindspore.tensor([0, 1], mindspore.int32)
>>> updates = mindspore.tensor([[0.5, 1.0], [1.0, 1.5]], mindspore.float32)
>>> mindspore.ops.inplace_index_add(var, indices, updates, axis=0)
>>> print(var.asnumpy())
[[1.5 3. ]
 [4.  5.5]
 [5.  6. ]]