torch.isnan — PyTorch 2.7 documentation (original) (raw)

• Docs >
• torch >
• torch.isnan
• 

Shortcuts

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 input is NaN and False elsewhere

Example:

torch.isnan(torch.tensor([1, float('nan'), 2])) tensor([False, True, False])

• torch.isnan
  • isnan()