PyTorch torch.isfinite 函数
Pytorch torch 参考手册torch.isfinite 是 PyTorch 中用于检查张量元素是否有有限值的函数。返回一个布尔张量，对于每个元素，如果是有限值（不是无穷大或NaN）则为True。
函数定义
torch.isfinite(input)
使用示例
实例
import torch
# 创建包含不同数值的张量
x = torch.tensor([1.0, float('inf'), float('nan'), 2.0])
# 检查是否有有限值
result = torch.isfinite(x)
print(result)
# 创建包含不同数值的张量
x = torch.tensor([1.0, float('inf'), float('nan'), 2.0])
# 检查是否有有限值
result = torch.isfinite(x)
print(result)
输出结果为：
tensor([True, False, False, True])