PyTorch torch.isfinite 函数

PyTorch torch.isfinite 函数 Pytorch torch 参考手册 torch.isfinite 是 PyTorch 中用于检查张量元素是否有有限值的函数。返回一个布尔张量，对于每个元素，如果是有限值（不是无穷大或NaN）则为True。函数定义 torch.isfinite(input) 使用示例实例 [mycode4 type='python'] import torc..

Updated 7/16/2026

Tutorial detail pages default to noindex.

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)

输出结果为：

tensor([True, False, False, True])

Pytorch torch 参考手册