torch.isnan¶
- torch.isnan(input) Tensor¶
Returns a new tensor with boolean elements representing if each element of
inputis NaN or not. Complex values are considered NaN when either their real and/or imaginary part is NaN.- Parameters:
input (Tensor) – the input tensor.
- Returns:
A boolean tensor that is True where
inputis NaN and False elsewhere
Example:
>>> torch.isnan(torch.tensor([1, float('nan'), 2])) tensor([False, True, False])
