torch.Tensor.masked_fill_

Tensor.masked_fill_(mask, value)

用 value 填充 self tensor 中 mask 為 True 的元素。mask 的形狀必須能與底層 tensor 的形狀進行廣播 (broadcastable)。

參數

• mask (BoolTensor) – 布林遮罩

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