mindspore.ops.MaskedFill

class mindspore.ops.MaskedFill

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

input 和 mask 的shape需相同或可广播。

输入：

• input (Tensor) - 输入Tensor，其数据类型为float16、float32、int8、或int32。

• mask (Tensor[bool]) - 输入的掩码，其数据类型为bool。

• value (Union[float, Tensor]) - 用来填充的值，只支持0维Tensor或float。

输出：

Tensor，输出与输入的数据类型和shape相同。

异常：

• TypeError - input 或 mask 不是Tensor。

• TypeError - value 既不是float也不是Tensor。

• TypeError - input 或 value 的数据类型不是float16、float32、int8、或int32。

• TypeError - value 的数据类型与 input 不同。

• TypeError - mask 的数据类型不是bool。

• ValueError - input 和 mask 的shape不可广播。

支持平台：

Ascend

样例：

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