c++ - 使用特征矩阵作为特征数组引用的参数

Question

我正在使用一个函数获取数组引用并更新它们的库:

void foo(ArrayXXd A&)

但是，在我的代码中我想使用

Matrix<double,Dynamic,Dynamic>

如何使用矩阵调用函数 foo？我能以某种方式将矩阵映射到数组吗？

这是编译器错误:

 error: invalid initialization of reference of type ‘Mat& {aka Eigen::Array<double, -1, -1>&}’ from expression of type ‘Eigen::Matrix<double, -1, -1>’

我做了以下似乎有效的，但我不知道它是否是一个通用的解决方案(不同的内存布局等)。

//X_IN is a Matrix<double,Dynamic,Dynamic> & 

//Map Matrix to pointer
X_pntr = X_IN.data();
//Map pointer to Array 
ArrayXXd X_array = Map<ArrayXXd>(X_pntr,X_IN.rows(),X_IN.cols());

foo(X_array);

Answer 1

Eigen 中的大多数对象都是表达式(更具体地说，是从 MatrixBase 派生的对象)。如果您想编写适用于任何类型的 Matrix/Array 等的函数，而不仅限于例如数组，需要写成如下形式:

template<typename T> 
void foo(Eigen::MatrixBase<T>& A)
{
   // do something here with A
}

现在您可以使用任何类型的对象调用 foo，例如 foo(A*A)，其中 A 是一个 MatrixXd，或 MatrixXf 或 ArrayXd，你明白了。更多细节请查看官方文档: http://eigen.tuxfamily.org/dox/TopicFunctionTakingEigenTypes.html

阅读您的评论后，我可以想出这个解决方案:

Eigen::Matrix<double,Eigen::Dynamic,Eigen::Dynamic> m(2,2);
m << 1,2,3,4;
Eigen::ArrayXXd tmp = m; // convert into array (via copy)
foo(tmp); // modify tmp
m = tmp; // copy back into m
cout << m; // now m is modified

当 Eigen 支持移动语义时，您将能够使用 std::move 而不是制作 2 个拷贝。