torch.isinf

torch.isinf(input) → 张量

测试input中的每个元素是否为无穷大（正无穷或负无穷）。

注意

当复数的实部或虚部为无穷大时，该复数为无穷大。

Parameters

输入 (张量) – 输入张量。

Returns

一个布尔张量，其中在input为无穷大的地方为True，其他地方为False

示例：

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