torch.Tensor.masked_fill_

Tensor.masked_fill_(mask, value)

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

引數

• mask (BoolTensor) – 布林掩碼

• value (float) – 用於填充的值