在Eigen中复制模板化函数参数

Question

我正在编写一个利用Eigen数据类型的泛型类。 我已经有问题将构造函数参数分配给类成员变量。 我的代码的简化版本是：

template <typename Derived>
class A
{
public:
  Eigen::Matrix<Derived> M; // error C2976: too few template parameters

A(const Eigen::DenseBase<Derived> & V)
{
  M = V.eval(); // I would want to snapshot the value of V.
}
};

我现在的问题是M应该是什么数据类型？ 我尝试了各种选项，例如：

Eigen::internal::plain_matrix_type_column_major<Derived> M;
Eigen::DenseBase<Derived> M;

但他们只会产生不同的错误。 请注意，我使用C ++ 17并期望从构造函数中推断出类模板参数。

Answer 1

Eigen::Matrix变量M的声明应该类似于：

Eigen::Matrix<typename Derived::Scalar, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor> M; 

然后你的代码将编译。 见演示 。

可以在此链接中找到每个模板参数的详细说明。

Answer 2

声明类型的一般方法是使用生成它作为源的声明，这样您就不必查看声明复杂模板类型的具体方法，这是基于您的代码的示例：

decltype(static_cast<Eigen::DenseBase<Derived> *>(nullptr)->eval()) M;

不要担心这里没有nullptr dereference，因为decltype中的代码没有被执行。

正如评论中指出的那样，有一种更清晰的写作方式：

decltype(declval<Eigen::DenseBase<Derived>>().eval()) M;

如果你担心这个类型可能是一个引用并且不想要：

remove_reference_t<decltype(declval<Eigen::DenseBase<Derived>>().eval())> M;

也不要忘记#include <type_traits>并使用std::前缀所有内容或using namespace std;添加using namespace std; 你的代码。

为了使语法更简单以供将来使用，请将其添加到代码的开头：

template<typename T, typename T::f>
using member_function_return_t = remove_reference_t<decltype(declval<T>().f())>;

而不是将变量声明为：

member_function_return_t<Eigen::DenseBase<Derived>, Eigen::DenseBase<Derived>::eval> M;

Answer 3

您的容器需要实际的“普通类型”作为模板参数：

template <typename PlainType>
class A
{
    PlainType M; 
public:
    template<class Derived>
    A(const Eigen::MatrixBase<Derived> & V) : M(V) {}
};

而且您还需要一个额外的模板扣除规则：

template<class Derived>
A(const Eigen::MatrixBase<Derived> & V) -> A<typename Derived::PlainObject>;

用法示例（ 关于godbolt ）：

template<class X>
void bar(X&); // just to read full type of A

void foo(Eigen::Matrix2d const& M)
{
    A a = M*M;
    bar(a);  // calls bar<A<Matrix2d>>();
}