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