Pytorch中的NaN

2020-05-15  本文已影响0人  SSRSGJYD

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

if torch.isnan(tensor).any():
   # do something
上一篇 下一篇

猜你喜欢

热点阅读