torch.Tensor.masked_fill_

Tensor.masked_fill_(mask, value)

當 mask 為 True 時，用 value 填充 self 張量中的元素。 mask 的形狀必須 可廣播 到底層張量的形狀。

引數

• mask (BoolTensor) – 布林掩碼

• value (float) – 要填充的值