mindspore.ops.inplace_update

查看源文件
mindspore.ops.inplace_update(x, v, indices)[源代码]

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

警告

这是一个实验性API，后续可能修改或删除。

说明

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

参数：

  • x (Tensor) - 待更新的Tensor。它可以是以下数据类型之一：float32、float16和int32。

  • v (Tensor) - 更新的Tensor，其类型与 x 相同，维度大小与 x 相同，但第一维度必须与 indices 的大小相同。

  • indices (Union[int, tuple[int], Tensor]) - 指定将 x 的哪些行更新为 v，由若干个整数构成。可以为int或tuple或1维Tensor，取值范围[- 'x' 的第一维, x 的第一维)。如果是tuple或Tensor，大小与v的第一个维度相同。

返回：

Tensor，更新后的Tensor，其类型和shape与 x 相同。

异常：

  • TypeError - indices 不是int或tuple或Tensor。

  • TypeError - indices 是tuple或Tensor，但是其中的元素不是int。

支持平台：

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)
>>> 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. ]]