mindspore.ops.MaskedFill

class mindspore.ops.MaskedFill[源代码]

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

更多参考详见 mindspore.ops.masked_fill()

支持平台:

Ascend GPU CPU

样例:

>>> input = Tensor(np.array([1., 2., 3., 4.]), mindspore.float32)
>>> mask = Tensor(np.array([True, True, False, True]), mindspore.bool_)
>>> output = ops.MaskedFill()(input, mask, 0.5)
>>> print(output)
[0.5 0.5 3.  0.5]