标量类型的特征模板

Question

我正在编写包含许多类的代码，这些类以标量类型为模板，T并在std::valarray<T>. 我想添加对 Eigen 容器的支持，但我无法找到一种仅在标量类型上进行模板化的方法，同时又保持了一些 Eigen 的灵活性。

例如，Eigen文档建议使用Eigen::DenseBase<Derived>，但这似乎意味着我必须将所有类定义更改为数组类型上的模板Derived，然后Derived::Scalar在任何地方使用。只是为了支持另一个容器而具有侵入性。

我也可以只使用Matrix<T,Dynamic,Dynamic>，但这是非常严格的。对于我关心的情况，我认为一个合理的中间立场是使用，Ref<Matrix<T,Dynamic,Dynamic>>因为它可以Matrix用一个界面覆盖和切片。但这似乎不起作用，我不确定为什么。这是我的意思的一个具体例子：

#include<iostream>
#include<Eigen/Dense>

using namespace std;
using namespace Eigen;

double
sum1(const Ref<const Matrix<double,Dynamic,Dynamic>> &m)
{
    double x = 0.0;
    for (int j=0; j<m.cols(); ++j)
        for (int i=0; i<m.rows(); ++i)
            x += m(i,j);
    return x;
}

template <typename T>
T
sum2(const Ref<const Matrix<T,Dynamic,Dynamic>> &m)
{
    T x = 0.0;
    for (int j=0; j<m.cols(); ++j)
        for (int i=0; i<m.rows(); ++i)
            x += m(i,j);
    return x;
}

int main()
{
    Matrix<double,Dynamic,Dynamic> a(2,2);
    a << 0,2,1,3;
    cout << "sum1(a) = " << sum1(a) << endl;            // ok
    cout << "sum2(a) = " << sum2(a) << endl;            // error
    cout << "sum2(a) = " << sum2<double>(a) << endl;    // ok
    return 0;
}

但是编译时出现错误：

$ clang++  -std=c++17 -I/usr/include/eigen3   eig2.cpp   -o eig2
eig2.cpp:33:29: error: no matching function for call to 'sum2'
    cout << "sum2(a) = " << sum2(a) << endl;            // error
                            ^~~~
eig2.cpp:19:1: note: candidate template ignored: could not match 'Ref' against
      'Matrix'
sum2(const Ref<const Matrix<T,Dynamic,Dynamic>> &m)
^
1 error generated.

有没有办法实现sum2它支持Matrix并且Ref不需要显式模板参数来使用它？

Answer 1

原因是

template <typename T>
T
sum2(const Ref<const Matrix<T,Dynamic,Dynamic>> &m)

不起作用的原因Matrix<T,Dynamic,Dynamic>是 C++ 不能推导模板参数并同时进行类型转换（Ref与 不同的类型Matrix，但它可以用 O(1) 的努力构造）。

允许您保留函数体的一种替代方法（需要 C++11 或更高版本）是：

template<typename Derived, typename T=typename Derived::Scalar>
T sum3(const MatrixBase<Derived> &m) 

如果你也想允许Array，你可以使用DenseBase，如果你想允许稀疏矩阵，你可以使用EigenBase。

但是，这并不等同于Ref使用显式类型替换的实现，因为 ifDerived是一个表达式，它可能会在您的函数中被多次评估。

另一种方法是写

template <typename T>
T
sum_impl(const Ref<const Matrix<T,Dynamic,Dynamic>> &m)
{ /* here comes the actual implementation */ }

template<typename Derived>
typename Derived::Scalar sum4(const Eigen::MatrixBase<Derived>& m)
{
    return sum_impl<typename Derived::Scalar>(m);
}