Pytorch操作检测NaNs

小猫的世界

是否有一个Pytorch-internal过程来检测张量中的NaNs ? Tensorflow 有 tf.is_nan 和tf.check_numerics操作
Pytorch也有类似的吗?但是好像我没有找到。我特别寻找Pytorch的内部例程，因为我希望这种情况发生在GPU和CPU上。
这不包括numpy（如：

1. np.isnan(sometensor.numpy()).any()

复制代码

）

2919005896

可以利用这个

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. >>> torch.isnan(x)
   
2. tensor([ 0,  0,  1], dtype=torch.uint8)

复制代码

莫晓沙

了解了，我们可以多看看别人的错误，让自己少犯错