torch.Tensor.masked_fill_

Tensor.masked_fill_(mask, value)

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

参数

• mask (BoolTensor) – 布尔掩码

• value (float) – 要填充的值