torch.Tensor.Tensor.masked_fill
Fills elements of self tensor with value where mask is True. Returns a new tensor with masked positions filled.
Parameters
maskTensor- Boolean mask tensor
valuenumber- Value to fill at masked positions
Returns
Tensor<S, D, Dev>– New tensor with masked positions filled