mindspore.ops.IndexFill

View Source On Gitee
class mindspore.ops.IndexFill[source]

Fills the elements under the dim dimension of the input Tensor x with the input value by selecting the indices in the order given in index.

Warning

This is an experimental API that is subject to change or deletion.

Refer to mindspore.ops.index_fill() for more details.

Inputs:

  • x (Tensor) - Input tensor.

  • dim (Union[int, Tensor]) - Dimension along which to fill the input tensor. Only supports a 0-dimensional tensor or an int number.

  • index (Tensor) - Indices of the input tensor to fill in.

  • value (Union[bool, int, float, Tensor]) - Value to fill the input tensor.

Outputs:

Tensor, has the same type and shape as input tensor.

Supported Platforms:

Ascend GPU CPU

Examples

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