mindspore.ops.InplaceUpdate

class mindspore.ops.InplaceUpdate(indices)

将 x 的特定行更新为 v 。

参数：

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

输入：

• x (Tensor) - 待更新的Tensor。数据类型支持float16、float32或int32。

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

输出：

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

异常：

• TypeError - indices 不是int或Tuple。

• TypeError - indices 为Tuple，而其包含的某一元素非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)
>>> inplace_update = ops.InplaceUpdate(indices)
>>> output = inplace_update(x, v)
>>> print(output)
[[0.5 1. ]
 [1.  1.5]
 [5.  6. ]]