Pytorch中的NaN
2020-05-15 本文已影响0人
SSRSGJYD
在Pytorch中,NaN与任何其他数进行比较都会返回False的结果。因此,在遇到数据范围的bug时,不能简单通过比较范围来判断数据是否正常,而是应该增加一个对NaN的判断:
if torch.isnan(tensor).any():
# do something
在Pytorch中,NaN与任何其他数进行比较都会返回False的结果。因此,在遇到数据范围的bug时,不能简单通过比较范围来判断数据是否正常,而是应该增加一个对NaN的判断:
if torch.isnan(tensor).any():
# do something