Torch C ++: API для проверки NAN

Question

Я использую libtorch C ++. В версии Python мы можем легко проверить значение тензора, вызвав его значение numpy, а в numpy мы имеем np.isnan(). Мне было интересно, есть ли встроенная функция в libtorch C++, чтобы проверить, имеет ли тензор какое-либо значение NAN?

Спасибо, Afshin

Answer 1

Попробуйте at::isnan.

int main() {
  torch::Tensor tensor = torch::rand({2, 3});
  std::cout << tensor << std::endl;
  std::cout << at::isnan(tensor) << std::endl;
  return 0;
}

Примечание: мне пришлось установить ночную сборку libtorch, так как стабильная версия не имела isnan.