torch.isinf¶
- 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])