torch.nonzero(torch.isnan(hidden_states)==True).size()
上述张量返回 [n, *],若n==0,无nan;n>0有nan
torch中查看某张量是否含有nan
最新推荐文章于 2024-10-29 16:56:36 发布
torch.nonzero(torch.isnan(hidden_states)==True).size()
上述张量返回 [n, *],若n==0,无nan;n>0有nan