torch.Tensor.masked_fill_¶
- Tensor.masked_fill_(mask, value)¶
Fills elements of
self
tensor withvalue
wheremask
is True. The shape ofmask
must be broadcastable with the shape of the underlying tensor.- Parameters
mask (BoolTensor) – the boolean mask
value (float) – the value to fill in with