时间戳

首页 > 解决方案 > 将 MatrixXd 的行传递给要修改的函数，而不在 Eigen 中创建副本

c++ - 将 MatrixXd 的行传递给要修改的函数，而不在 Eigen 中创建副本

问题描述

这个问题与 C++ 和 Eigen 有关。

我有一个函数可以计算一些向量Eigen::VectorXd并将它们存储到作为参数传递的向量中。函数的返回值是函数中完成的迭代次数。下面是函数的伪代码：

size_t integrate(Eigen::VectorXd& v) {
    size_t numIterations = 0;
    // do some iterative calculations and increment numIterations
    // vNew is generated from the new calculations
    v = vNew;
    return numIterations;
}

我想使用这个功能的方式如下：

Eigen::MatrixXd q = Eigen::MatrixXd::Zero(numRows, numCols);

for(size_t i = 0; i < numSteps; i++) {
    integrate(q.row(i));
}

但是，这不会在 Visual Studio 中编译。我收到C++ initial value of reference to non-const must be an lvalue编译器错误。我的目的是避免不必要的复制，并在 for 循环的每次迭代中将计算结果（向量）放入矩阵中。我知道这row()不会返回一个VectorXd. 我已经尝试了一些其他的东西，但都没有奏效。MatrixXd填充Eigen中预先分配的行的有效方法是什么？

注意：我知道我可以通过 ref 将矩阵连同当前值i和 do q.row(i) = vNew;in一起传递给函数integrate()。然而，它看起来很丑陋，对我来说是不必要的。integrate()真的不需要知道任何关于 a 的事情MatrixXd。

谢谢你。

标签： c++eigen

解决方案

您可以使用Eigen::Ref对象将可写引用传递给矩阵块。在您的情况下，由于（主要列）矩阵的行将具有内部跨度，因此您需要明确允许：

size_t integrate(Eigen::Ref<Eigen::VectorXd, 0, Eigen::InnerStride<> > v) {
    v.setRandom(); // use `v` as if it was a VectorXd
    return 0;
}

为了获得更好的性能，您应该考虑q以行主要形式存储，或者将各个列传递给integrate：

Eigen::Matrix<double, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor> q; q.setZero(numRows, numCols);
// ...
integrate(q.row(i));

或者：

Eigen::MatrixXd q; q.setZero(numRows, numCols); // you'll need to switch how many rows and cols are to be allocated
// ...
integrate(q.col(i));

在任何一种情况下，integrate都可以采用简单的Eigen::Ref<Eigen::VectorXd>（又名Eigen::Ref<Eigen::VectorXd, 0, Eigen::InnerStride<1> >）：

size_t integrate(Eigen::Ref<Eigen::VectorXd> v) {
    // ...

推荐阅读