torch.Tensor.masked_fill_

Tensor.masked_fill_(mask, value)

将 self 张量的元素填充为 value ，其中 mask 为 True。 mask 的形状必须与底层张量的形状可广播。

参数:

• 布尔掩码（BoolTensor）-布尔掩码

• 值（float）-要填充的值