torch7：过滤掉 NaN 值

Question

给定任何一般 float torch.Tensor，可能包含一些 NaN 值，我正在寻找一种有效的方法来将其中的所有 NaN 值替换为零，或者完全删除它们并过滤掉另一个新 Tensor 中的 "useful" 值。

我知道一个简单的方法是手动遍历给定张量中的所有值（并相应地将它们替换为零或拒绝它们用于新张量）。

是否有一些预定义的 Torch 函数或函数组合可以在性能方面更有效地实现这一点，这依赖于 Torch 固有的 CPU-GPU 优化？

Answer 1

嗯，torch 中似乎没有检查 NaN 张量的功能。但是因为 NaN != NaN，所以有一个解决方法：

a = torch.rand(4, 5)
a[2][3] = tonumber('nan')
nan_mask = a:ne(a)
notnan_mask = a:eq(a)

print(a)
 0.2434  0.1731  0.3440  0.3340  0.0519
 0.0932  0.4067  nan     0.1827  0.5945
 0.3020  0.1035  0.5415  0.3329  0.7881
 0.6108  0.9498  0.0406  0.9335  0.3582
[torch.DoubleTensor of size 4x5]

print(nan_mask)
 0  0  0  0  0
 0  0  1  0  0
 0  0  0  0  0
 0  0  0  0  0
[torch.ByteTensor of size 4x5]

有了这些掩码，您可以有效地提取 NaN/not NaN 值并将它们替换为您想要的任何值：

print(a[notnan_mask])
...
[torch.DoubleTensor of size 19]

a[nan_mask] = 42
print(a)
  0.2434   0.1731   0.3440   0.3340   0.0519
  0.0932   0.4067  42.0000   0.1827   0.5945
  0.3020   0.1035   0.5415   0.3329   0.7881
  0.6108   0.9498   0.0406   0.9335   0.3582
[torch.DoubleTensor of size 4x5]

torch7：过滤掉 NaN 值

torch7: Filtering out NaN values

lua

nan

torch