torch.Tensor.masked_fill_

Tensor.masked_fill_(mask, value)

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

Parameters

• mask (BoolTensor) – 布尔掩码

• 值 (浮点数) – 要填充的值