mindspore.ops.masked_fill

mindspore.ops.masked_fill(input_x, mask, value)

在掩码为 True 位置填充指定值。

支持广播。

参数：

• input_x (Tensor) - 输入tensor。

• mask (Tensor[bool]) - 输入掩码。

• value (Union[Number, Tensor]) - 指定值。

返回：

Tensor

支持平台：

Ascend GPU CPU

样例：

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