torch.Tensor.masked_fill

Tensor.masked_fill(mask, value) → Tensor

Out-of-place version of torch.Tensor.masked_fill_()