满足条件

Question

我希望从符合条件的密集矩阵中获取行/列索引。在我的情况下，结果可能非常稀少。例如，使用矩阵

1 5 2
7 6 3
2 3 8

我想获得系数大于4或

的指标

(0,1), (1,0), (1,1), (2,2)

我最初的想法包括使用select或coefficientwise操作来构建bool / int矩阵

0 1 0
1 1 0
0 0 1

然后将其转换为稀疏矩阵

(0,1,1), (1,0,1), (1,1,1), (2,2,1)

然后删除coeff值

(0,1), (1,0), (1,1), (2,2)

然而，这需要两次遍历原始矩阵大小的矩阵，这可能非常大。

或者，在原始矩阵上的原始双循环类似于伪代码

for (int i; i < mat.cols(); i++) {
    for (int j; j < mat.rows(); j++) {
        if(cond(mat(j, i))) {
            add_index(i, j, index_list)
        }
    }
}

但这只需要编译器优化，而不需要Eigen的优化/矢量化。

我缺少一种更有效的方法吗？在我的情况下，条件是简单的比较。

感谢您的帮助

Answer 1

这里没有太多的矢量化，但是为了避免编写两个循环（最好可以在行主存储中反转），最好的工具就是访问者：

mat.visit(some_visitor);

可悲的是，访客不能是一个简单的lambda，因为visit为第一个元素调用方法init(val,0,0)。这对于类似减少的访问者很有用，但并非总是如此。为了使visit接受一个简单的lambda，您可以使用以下帮助器：

template<typename Func>
struct lambda_as_visitor_wrapper : Func {
    lambda_as_visitor_wrapper(const Func& f) : Func(f) {}
    template<typename S,typename I>
    void init(const S& v, I i, I j) { return Func::operator()(v,i,j); }
};

template<typename Mat, typename Func>
void visit_lambda(const Mat& m, const Func& f)
{
    lambda_as_visitor_wrapper<Func> visitor(f);
    m.visit(visitor);
}

在您的情况下，您可以写：

int main() {
    int n = 5;
    double th = 0.5;
    Eigen::MatrixXd M = Eigen::MatrixXd::Random(n,n);

    std::vector<std::pair<int,int>> indices;
    visit_lambda(M,
        [&indices,th](double v, int i, int j) {
            if(v>th)
                indices.push_back(std::make_pair(i,j));
        });


    std::cout << M << "\n\n";

    for(auto p:indices)
        std::cout << '(' << p.first << ',' << p.second << ") ";
    std::cout << '\n';

    return 0;
}