c++ - 如何计算 eigen::Tensor 中非零的数量

Question

如果我用bool张量求和，结果永远是1

Eigen::Tensor<int,2> my_rankx (3,3);
my_rankx.setValues(
{
    {1, 2, 3},
    {0, 0, 0},
    {11, 12, 0}
});

auto number_matrix = (my_rankx.constant(static_cast<int>(0))==my_rankx);
cout<<number_matrix.sum()<<endl; // result of cout is 1

Answer 1

(my_rankx.constant(0) == my_rankx)是 bool 的张量s，并添加 bool s 一起导致 false或 true , 转换为 0和 1分别。 (顺便说一句，问题中的 static_cast<int>(0) 是多余的，因为 0 是 int 类型。)

您可以使用 std::count 连同 data() 相反:

std::cout << std::count(my_rankx.data(), my_rankx.data() + my_rankx.size(), 0)
          << "\n";