Rate this Page

torch.Tensor.masked_fill_#

Tensor.masked_fill_(mask, value)#

Fills elements of self tensor with value where mask is True. The shape of mask must be broadcastable with the shape of the underlying tensor.

Parameters
  • mask (BoolTensor) – the boolean mask

  • value (float) – the value to fill in with