mindspore.Tensor.masked_fill

mindspore.Tensor.masked_fill(mask, value)

将掩码位置为True的位置填充指定的值。此tensor和 mask 的shape需相同或可广播。

参数：

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

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

返回：

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

异常：

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

• TypeError - mask 不是Tensor。

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

• TypeError - 此tensor或 value 的数据类型不是bool, int8, int32, int64, float16, float32, bfloat16。

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

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

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor, ops
>>> input_x = Tensor(np.array([1., 2., 3., 4.]), mindspore.float32)
>>> mask = Tensor(np.array([True, True, False, True]), mindspore.bool_)
>>> output = Tensor.masked_fill(input_x, mask, 0.5)  #input_x.masked_fill(mask, 0.5)
>>> print(output)
[0.5 0.5 3.  0.5]