釉她 发表于 2025-7-21 16:30:23

关于nan的大小判定规则

技术背景

在Numpy或PyTorch框架编程的过程中,如果有计算异常,可能会存在NAN这样的一个东西,也就是not a number。这不是一个常规的数字,但是也可以参与计算,例如加和、比较等等。本文通过一个简单的示例,展示一下PyTorch框架下,NAN的计算规则。
代码示例

如下是一个简单的示例:
In : import torch as tc

In : tc.nan
Out: nan

In : a = tc.tensor()

In : a
Out: tensor()

In : a.max()
Out: tensor(nan)

In : a.min()
Out: tensor(nan)

In : a.sum()
Out: tensor(nan)

In : b = a - tc.nan

In : b
Out: tensor()我们可以直接用torch.nan来定义一个NAN的数。然后这个NAN在计算过程中,不论是取最大值还是最小值,都会取到NAN。如果另外一个数跟NAN做四则运算,得到的结果也是NAN,这就是NAN的基本运算规则。
NAN的产生

如果我们直接使用1/0这样的形式去计算,得到的结果会是inf而不是NAN:
In : import torch as tc

In : a = tc.zeros(1)

In : b = tc.ones(1)

In : c=b/a

In : c
Out: tensor()实际上出现NAN时,只有两种可能的情况:0/0和inf/inf:
In : import torch as tc

In : a = tc.zeros(1)

In : b = tc.zeros(1)

In : a/b
Out: tensor()

In : b=1

In : a/b
Out: tensor()

In : b/a
Out: tensor()

In : (b/a)/(b/a)
Out: tensor()那么就可以通过这样的规则,去排除一下是计算的哪个步骤出现了问题。
总结概要

本文通过几个简单的代码示例,展示了一下NAN在PyTorch框架下形成的原因。通过了解这个原因和规则,有助于解决在深度学习开发和训练过程中出现的NAN的问题。
版权声明

本文首发链接为:https://www.cnblogs.com/dechinphy/p/tc-nan.html
作者ID:DechinPhy
更多原著文章请参考:https://www.cnblogs.com/dechinphy/
打赏专用链接:https://www.cnblogs.com/dechinphy/gallery/image/379634.html
腾讯云专栏同步:https://cloud.tencent.com/developer/column/91958

来源:程序园用户自行投稿发布,如果侵权,请联系站长删除
免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!
页: [1]
查看完整版本: 关于nan的大小判定规则