pytorch - 手电筒 C++ : API to check NAN

Question

我正在使用 libtorch C++。在 python 版本中，我们可以通过调用其 numpy 值轻松检查张量的值，而在 numpy 中，我们有 np.isnan()。我想知道 libtorch C++ 中是否有内置函数来检查张量是否有任何 NAN 值？

谢谢，阿夫欣

Answer 1

添加 Fábio 的回答(我的声誉太低无法发表评论):

如果您真的想在 assert 或 if 条件中使用有关 NAN 的信息，您需要将其从 torch::Tensor 转换而来像这样的 C++ bool

torch::Tensor myTensor;
// do something
auto tensorIsNan = at::isnan(myTensor).any().item<bool>(); // will be of type bool