自己写神经网络和损失函数,真的非常容易出现nan值。
判断方法:变量==变量
import torch
a=torch.rand(3,4)
a[0][1]=torch.nan
print(a==a)
然后,它的结果是:
在nan处结果不相等,是False。这就达到了判断目的。
自己写神经网络和损失函数,真的非常容易出现nan值。
判断方法:变量==变量
import torch
a=torch.rand(3,4)
a[0][1]=torch.nan
print(a==a)
然后,它的结果是:
在nan处结果不相等,是False。这就达到了判断目的。