torch中查看某张量是否含有nan

torch.nonzero(torch.isnan(hidden_states)==True).size()
上述张量返回 [n, *],若n==0,无nan;n>0有nan

你可能感兴趣的:(pytorch)