mindspore.Tensor.inplace_update

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

根据 indicesv 来更新Tensor中的值。

Note

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

参数:
  • v (Tensor) - 用来更新的值。有相同的数据类型和除第一维外相同的shape。第一维的大小应该与 indices 大小相同。

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

返回:

Tensor,更新后的Tensor。

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

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

  • ValueError - Tensor的shape与 v 的shape不同。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> import mindspore
>>> x = Tensor(np.array([[1, 2], [3, 4], [5, 6]]), mindspore.float32)
>>> v = Tensor(np.array([[0.1, 0.2], [0.3, 0.4]]), mindspore.float32)
>>> indices = (0, 1)
>>> output = x.inplace_update(v, indices)
>>> print(output)
[[0.1 0.2]
 [0.3 0.4]
 [5.  6. ]]