torch.Tensor.masked_fill_

Tensor.masked_fill_(mask, value)

使用value填充self张量中mask为True的元素。同时，mask的形状必须与底层张量的形状兼容进行广播操作。

参数

• mask (BoolTensor) – 布尔类型的掩码

• value (float) — 填充的值