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])