检测 NaN 的 Pytorch 操作

Question

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

我正在专门寻找 Pytorch 内部例程，因为我希望这发生在 GPU 和 CPU 上。这不包括基于 numpy 的解决方案(如 np.isnan(sometensor.numpy()).any() )...

Answer 1

您可以随时利用 nan != nan 的事实。 :

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

pytorch 0.4 还有 torch.isnan :

>>> torch.isnan(x)
tensor([ 0,  0,  1], dtype=torch.uint8)