mindspore.ops.inplace_update

mindspore.ops.inplace_update(x, v, indices)[源代码]

根据 indices,将 x 中的某些值更新为 v

说明

indices 只能沿着最高轴进行索引。

参数:
  • x (Tensor) - 待更新的Tensor。

  • v (Tensor) - 更新的值。

  • indices (Union[int, tuple], Tensor) - 待更新值在原Tensor中的索引。

返回:

Tensor,更新后的Tensor。

异常:
  • TypeError - indices 不是int或tuple。

  • TypeError - indices 是元组,但是其中的元素不是int。

支持平台:

Ascend GPU CPU

样例:

>>> indices = (0, 1)
>>> x = Tensor(np.array([[1, 2], [3, 4], [5, 6]]), mindspore.float32)
>>> v = Tensor(np.array([[0.5, 1.0], [1.0, 1.5]]), mindspore.float32)
>>> output = ops.inplace_update(x, v, indices)
>>> print(output)
[[0.5 1. ]
 [1.  1.5]
 [5.  6. ]]