pytorch torch.isclose函数介绍

torch.isclose 是 PyTorch 中用于比较两个张量是否"近似相等"的函数。它主要用于判断两个张量的对应元素在数值上是否接近(考虑了浮点数精度的可能误差)。


函数定义

复制代码
torch.isclose(input, other, rtol=1e-05, atol=1e-08, equal_nan=False)

参数说明

  1. input (Tensor):

    • 第一个张量。
  2. other (Tensor):

    • 第二个张量,和 input 的形状必须相同,或者可以通过广播机制与 input 对齐。
  3. rtol (float, 可选,默认值:1e-05):

    • 相对容忍误差(relative tolerance)。比较时的相对误差阈值,定义了两个值相对距离的可接受范围。
  4. atol (float, 可选,默认值:1e-08):

    • 绝对容忍误差(absolute tolerance)。比较时的绝对误差阈值,定义了两个值绝对距离的可接受范围。
  5. equal_nan (bool, 可选,默认值:False):

    • 是否将 NaN 视为"接近"。
    • 如果为 True,则两个 NaN 会被认为是相等的。

返回值

  • 返回一个与 inputother 形状相同的布尔张量。
  • 每个元素表示 inputother 对应位置的元素是否"近似相等"。

比较规则

两个元素 aa 和 bb 被认为是"近似相等"的条件是:

  • |a - b|: 表示 inputother 对应元素之间的绝对差值。
  • atol: 绝对误差阈值。
  • rtol: 相对误差阈值。

常见用途

  • 比较浮点数是否相等(避免浮点数精度误差)。
  • 检查数值计算中张量的结果是否一致或接近。
  • 判断两个张量之间的元素是否在某个容忍范围内。

示例

复制代码
import torch

# 创建两个浮点张量
a = torch.tensor([1.0, 2.0, 3.0001])
b = torch.tensor([1.0, 2.0, 3.0])

# 默认参数下比较
result = torch.isclose(a, b)
print(result)  # tensor([True, True, False])

# 调整容忍误差
result = torch.isclose(a, b, rtol=1e-03, atol=1e-05)
print(result)  # tensor([True, True, True])

# 比较含有 NaN 的张量
a = torch.tensor([1.0, float('nan')])
b = torch.tensor([1.0, float('nan')])

# 默认不认为 NaN 相等
print(torch.isclose(a, b))  # tensor([ True, False])

# 允许 NaN 相等
print(torch.isclose(a, b, equal_nan=True))  # tensor([ True, True])

注意事项

  • rtolatol 的选择

    • 如果数值范围较大,可以增加 rtol
    • 如果数值精度要求较高,可以减小 atol
  • NaN 的比较

    • 默认情况下,torch.isclose 认为 NaN 是不相等的,除非 equal_nan=True
相关推荐
AI人工智能+19 分钟前
表格识别技术:通过计算机视觉和OCR,实现非结构化表格向结构化数据的转换,推动数字化转型。
人工智能·计算机视觉·ocr
小宁爱Python36 分钟前
基于 Django+Vue3 的 AI 海报生成平台开发(海报模块专项)
人工智能·python·django
破烂儿38 分钟前
基于机器学习的缓存准入策略研究
人工智能·机器学习·缓存
红豆怪怪43 分钟前
[LeetCode 热题 100] 32. 最长有效括号
数据结构·python·算法·leetcode·动态规划·代理模式
算法打盹中44 分钟前
SimLingo:纯视觉框架下的自动驾驶视觉 - 语言 - 动作融合模型
人工智能·机器学习·计算机视觉·语言模型·自动驾驶
大嘴带你水论文2 小时前
震惊!仅用10张照片就能随意编辑3D人脸?韩国KAIST最新黑科技FFaceNeRF解析!
论文阅读·人工智能·python·科技·计算机视觉·3d·transformer
CodeCraft Studio2 小时前
国产化PDF处理控件Spire.PDF教程:如何在 Java 中通过模板生成 PDF
java·python·pdf·spire.pdf·java创建pdf·从html创建pdf
IT_陈寒2 小时前
🔥3分钟掌握JavaScript性能优化:从V8引擎原理到5个实战提速技巧
前端·人工智能·后端
格林威2 小时前
棱镜的技术加持:线扫相机如何同时拍RGB和SWIR?
人工智能·深度学习·数码相机·yolo·计算机视觉
JoinApper2 小时前
小白学OpenCV系列3-图像算数运算
人工智能·opencv·计算机视觉