Pytorch操作检测NaNs

fantomas

是否有Pytorch内部程序来检测张量中的NaN？ Tensorflow有tf.is_nan和tf.check_numerics操作...... Pytorch在某处有类似的东西吗？我在文档中找不到这样的东西......

fantomas

可以利用这个

1. nan != nan:

复制代码

1. >>> x = torch.tensor([1, 2, np.nan])
   
2. tensor([  1.,   2., nan.])
   
3. >>> x != x
   
4. tensor([ 0,  0,  1], dtype=torch.uint8)

复制代码

使用pytorch 0.4还有torch.isnan：

1. 使用pytorch 0.4还有torch.isnan：

复制代码