mindspore.ops.masked_fill

mindspore.ops.masked_fill(input_x, mask, value)[source]

Fills elements of tensor with value where mask is True.

Support broadcast.

Parameters

input_x (Tensor) – The input tensor.
mask (Tensor[bool]) – The input mask.
value (Union[Number, Tensor]) – The value to fill in with.

Returns

Tensor

Supported Platforms:: Ascend GPU CPU

Examples

>>> import mindspore
>>> input_x = mindspore.tensor([1., 2., 3., 4.], mindspore.float32)
>>> mask = mindspore.tensor([True, True, False, True], mindspore.bool_)
>>> output = mindspore.ops.masked_fill(input_x, mask, 0.5)
>>> print(output)
[0.5 0.5 3.  0.5]