torch.Tensor.Tensor.masked_fill_
In-place version of masked_fill. Fills elements of self tensor with value where mask is True.
Parameters
maskTensor- Boolean mask tensor
valuenumber- Value to fill at masked positions
Returns
this– This tensor modified in-place