torch.isinf

torch.isinf(input) → Tensor

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

注意

复数值在其实部或虚部为无穷大时为无穷大。

参数:

input (Tensor) – 输入张量。

返回值:

一个布尔张量，当 input 为无穷大时为 True，其他地方为 False

示例：

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