mindspore.ops.inplace_sub

mindspore.ops.inplace_sub(x, v, indices)

根据索引将 x 减 v 。

对于 indices 的每个元素下标 $i,...,j$ ：

$$x[\text{index}[i,...,j]]-=v[i,...,j]$$

参数：

• x (Tensor) - 输入tensor。

• v (Tensor) - 被 x 减去的tensor。

• indices (Union[int, tuple]) - 对输入 x 沿第0维度的索引。

返回：

Tensor

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> import numpy as np
>>> indices = (0, 1)
>>> x = mindspore.tensor(np.array([[1, 2], [3, 4], [5, 6]]), mindspore.float32)
>>> input_v = mindspore.tensor(np.array([[0.5, 1.0], [1.0, 1.5]]), mindspore.float32)
>>> output = mindspore.ops.inplace_sub(x, input_v, indices)
>>> print(output)
[[0.5 1. ]
 [2.  2.5]
 [5.  6. ]]