torch数组计算出现NAN

在torch中零除零不会报错,结果会是NaN。

比如

a = torch.tensor([0,0])

a/a

output: tensor([nan])

PS: 这个白痴问题我想了好久,最后还是靠print才解决。没想到这里是因为平常如果除零的话是会报错的,但是在numpy里会提醒,torch里不提醒。解决办法就是a.clamp(1e-4)