torch.isfinite

torch.isfinite(input) → Tensor

返回一个新张量，其中包含布尔元素，表示每个元素是否为 有限 值。

实数值在不是 NaN、负无穷大或无穷大时为有限值。复数值在其真实和虚部均为有限值时为有限值。

参数

input (Tensor) – 输入张量。

返回值

一个布尔张量，如果 input 是有限值，则为 True，否则为 False

示例

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