mindspore.ops.index_fill

View Source On Gitee
mindspore.ops.index_fill(x, axis, index, value)[source]

Fills the elements of the input x with value along the given axis and indices.

Parameters

  • x (Tensor) – The input tensor.

  • axis (Union[int, Tensor]) – The specified axis.

  • index (Tensor) – The specified indices.

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

Returns

Tensor

Supported Platforms:

Ascend GPU CPU

Examples

>>> import mindspore
>>> x = mindspore.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]], mindspore.float32)
>>> index = mindspore.tensor([0, 2], mindspore.int32)
>>> value = mindspore.tensor(-2.0, mindspore.float32)
>>> y = mindspore.ops.index_fill(x, 1, index, value)
>>> print(y)
[[-2. 2. -2.]
 [-2. 5. -2.]
 [-2. 8. -2.]]