mindspore.ops.IndexPut

查看源文件
class mindspore.ops.IndexPut(accumulate=0)[源代码]

根据 indices 中的下标值,使用 x2 中的数值替换 x1 中的相应元素的值。

参数:
  • accumulate (int) - 如果 accumulate 被设置为1 , x2 中的元素被累加到 x1 的相应元素上; 如果为 0,用 x2 中的元素取代 x2 的相应元素。默认值: 0

输入:
  • x1 (Tensor) - 被执行替换操作的Tensor, 维度大于等于1。

  • x2 (Tensor) - 数据类型和 x1 一致的一维的Tensor。如果其size为1,则shape将被广播为 x1 的shape。

  • indices (tuple[Tensor], list[Tensor]) - 元素类型是int32或者int64, 用于对 x1 中的元素进行索引。 indices 中的tensor的秩应为1-D, indices 中tensor的size应小于 x1 的秩,indices中的tensor应是可广播的。

输出:

Tensor, 其数据类型和shape与 x1 相同。

异常:
  • TypeError - 如果 x1 的dtype与 x2 的dtype不同。

  • TypeError - 如果 indices 不是tuple[Tensor]或者list[Tensor]。

  • TypeError - 如果 indices 中的tensor的dtype不是int32或者int64。

  • TypeError - 如果 indices 中的tensor的dtype是不一致的。

  • TypeError - 如果 accumulate 的dtype不是int。

  • ValueError - 如果 x2 的秩不是1-D。

  • ValueError - 当 x1 的rank与 indices 的size相等时,如果 x2 的size不为1 或者不为 indices 中Tensor的最大size。

  • ValueError - 当 x1 的rank大于 indices 的size时,如果 x2 的size不为1 或者不为 x1 的最后一维的shape。

  • ValueError - 如果 indices 中的tensor的秩不是1-D。

  • ValueError - 如果 indices 中的tensor不是可广播的。

  • ValueError - 如果 indices 的size大于 x1 的秩。

  • ValueError - 如果 accumulate 的值不是0或1。

支持平台:

Ascend CPU

样例:

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor, ops
>>> x1 = Tensor(np.array([[1, 2, 3], [4, 5, 6]]).astype(np.int32))
>>> x2 = Tensor(np.array([3]).astype(np.int32))
>>> indices = [Tensor(np.array([0, 0]).astype(np.int32)), Tensor(np.array([0, 1]).astype(np.int32))]
>>> accumulate = 1
>>> op = ops.IndexPut(accumulate = accumulate)
>>> output = op(x1, x2, indices)
>>> print(output)
 [[4 5 3]
 [4 5 6]]