indexing - Xtensor 返回值为 NaN 的索引

Question

假设我有一个张量：

xt::xtensor_fixed<float, xt::xshape<1, 4>, xt::layout_type::column_major> withnans =
 { {3, std::numeric_limits<double>::quiet_NaN(), 4, std::numeric_limits<double>::quiet_NaN()} };

即{{3, nan, 4, nan}}

我想返回有 nan 的位置（索引）。

{1, 3}

我将如何在 xtensor 中执行此操作？

Answer 1

没关系。

xt::xtensor_fixed<float, xt::xshape<1, 4>, xt::layout_type::column_major> withnans =
 { {3, std::numeric_limits<double>::quiet_NaN(), 4, std::numeric_limits<double>::quiet_NaN()} };

std::cout << xt::flatten_indices(xt::argwhere(xt::isnan(xt::flatten(withnans)))) << std::endl;

{3}