如何計算特征::張量中的非零數<T,2>

Question

如果我使用布爾張量的總和，則重用永遠是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的張量，將bool加在一起會導致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";