美文网首页
Pytorch中的NaN

Pytorch中的NaN

作者: SSRSGJYD | 来源:发表于2020-05-15 11:55 被阅读0次

在Pytorch中,NaN与任何其他数进行比较都会返回False的结果。因此,在遇到数据范围的bug时,不能简单通过比较范围来判断数据是否正常,而是应该增加一个对NaN的判断:

if torch.isnan(tensor).any():
   # do something

相关文章

网友评论

      本文标题:Pytorch中的NaN

      本文链接:https://www.haomeiwen.com/subject/jdfcohtx.html