特征模板 function 接受矩陣 col 和向量

Question

我不明白如何編寫同時接受向量和矩陣列的模板化 function？

例如：

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); 沒關系。

我見過使用預定義類型執行此操作的示例，也見過探索模板的示例。 他們都沒有展示如何使用模板化 function 解決所描述的任務。

編輯：我也想傳遞動態大小向量，但不是必須的，固定大小

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);
}