mindspore.ops.IndexFill

class mindspore.ops.IndexFill[源代码]

按 index 中给定的顺序选择索引，将输入Tensor x 的 dim 维度下的元素用 value 的值填充。

警告

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

更多参考详见 mindspore.ops.index_fill()。

输入：

  • x (Tensor) - 输入Tensor。

  • dim (Union[int, Tensor]) - 填充输入Tensor的维度，要求是一个int或者数据类型为int32或int64的零维Tensor。

  • index (Tensor) - 填充输入Tensor的索引，数据类型为int32。

  • value (Union[bool, int, float, Tensor]) - 填充输入Tensor的值。

输出：

填充后的Tensor。shape和数据类型与输入 x 相同。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor, ops
>>> index_fill = ops.IndexFill()
>>> x = Tensor(np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]]).astype(np.float32))
>>> index = Tensor([0, 2], mindspore.int32)
>>> value = Tensor(-2.0, mindspore.float32)
>>> y = index_fill(x, 1, index, value)
>>> print(y)
[[-2. 2. -2.]
 [-2. 5. -2.]
 [-2. 8. -2.]]