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]