torch.isinf — PyTorch 2.0 documentation (original) (raw)

• Docs >
• torch >
• torch.isinf
• 

Shortcuts

torch.isinf(input) → Tensor¶

Tests if each element of input is infinite (positive or negative infinity) or not.

Note

Complex values are infinite when their real or imaginary part is infinite.

Parameters:

input (Tensor) – the input tensor.

Returns:

A boolean tensor that is True where input is infinite and False elsewhere

Example:

torch.isinf(torch.tensor([1, float('inf'), 2, float('-inf'), float('nan')])) tensor([False, True, False, True, False])

• torch.isinf