c++ - 接受矩阵 col 和向量的特征模板函数

Question

我不明白如何编写同时接受向量和矩阵列的模板化函数？

例如：

template<typename T>
void foo(
    const Eigen::MatrixX<T>& M){

}

int main(){
  Eigen::VectorX<double> v(3);
  Eigen::MatrixX<double> m(4,3);

  foo(m); // fine
  foo(m.col(0)); // broken
  foo(m.row(0)); // broken
  foo(v); // broken
}

只有foo(m);没问题。

我见过使用预定义类型执行此操作的示例，也见过探索模板的示例。他们都没有展示如何使用模板函数解决所描述的任务。

编辑： 我也想传递动态大小向量，但不是必须的，固定大小

标签： c++eigeneigen3

Answer 1

我可以使用MatrixBase：

#include <Eigen/Dense>

template<typename T>
void foo(const Eigen::MatrixBase<T>& M){}

int main(){
    Eigen::Vector3d v(3);
    Eigen::Matrix<double,4,3> m(4,3);
    Eigen::Matrix<double,Eigen::Dynamic,Eigen::Dynamic> q(5,6);

    foo(m);
    foo(m.col(0));
    foo(m.row(0));
    foo(v);
    foo(q);
}