Rate this Page
★★★★★
torch.isinf#
- torch.isinf(input)→Tensor#
Tests if each element of
inputis infinite(positive or negative infinity) or not.Note
Complex values are infinite when their real or imaginary part isinfinite.
- Parameters
input (Tensor) – the input tensor.
- Returns
A boolean tensor that is True where
inputis infinite and False elsewhere
Example:
>>>torch.isinf(torch.tensor([1,float('inf'),2,float('-inf'),float('nan')]))tensor([False, True, False, True, False])
On this page
