Pytorch中的NaN

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

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

你可能感兴趣的:(Pytorch中的NaN)