mindspore.ops.InplaceAdd

class mindspore.ops.InplaceAdd(indices)

将 input_v 添加到 x 的特定行。即计算 y = x; y[i,] += input_v 。

更多参考详见 mindspore.ops.inplace_add()。

参数：

• indices (Union[int, tuple]) - 指定将 input_v 添加到 x 的哪些行。可以为int或Tuple，取值范围[0, len(x)]。

输入：

• x (Tensor) - 输入Tensor，shape为：\((N,*)\) ，其中 \(*\) 表示任何数量的附加维度。

• input_v (Tensor) - 添加到 x 的Tensor。除第一个维度之外shape必须与 x 的shape相同。第一个维度必须与 indices 的长度相同。数据类型与 x 相同。

输出：

Tensor，与 x 的shape和数据类型相同。

支持平台：

Ascend GPU CPU

样例：

>>> import numpy as np
>>> import mindspore
>>> from mindspore import Tensor, ops
>>> indices = (0, 1)
>>> x = Tensor(np.array([[1, 2], [3, 4], [5, 6]]), mindspore.float32)
>>> input_v = Tensor(np.array([[0.5, 1.0], [1.0, 1.5]]), mindspore.float32)
>>> inplaceAdd = ops.InplaceAdd(indices)
>>> output = inplaceAdd(x, input_v)
>>> print(output)
[[1.5 3. ]
 [4.  5.5]
 [5.  6. ]]