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