为特征向量和std：：向量编写模板函数

Question

我希望编写一个模板函数来接受Eigen:：vectrox*（float/int/double）和std:：vector

如何声明模板？下面的方法不起作用。

// header.h

//std::vector function
template <typename T, typename A>
void do_something(std::vector<T, A> &stdvec)
{
    //do stuff with std::vector<T>
    //e.g. std::fill(stdvec.begin(),stdvec.end(),0);
}

//Eigen vector function
template <typename Derived>
void do_something(Eigen::EigenBase<Derived> &evec)
{
    //do stuff with Eigen::VectorX
    //e.g. evec.setZero()
}

//single value function
template <typename T>
void do_something(T &var)
{
    //do stuff with int/float/double
    //e.g. var = 0;
}

其基本原理是不编写单独的函数所有不同的组合。

Answer 1

诀窍不是编写几个函数（如您所发现的），而是应用标量函数（valdo_something（T

do_something(var);

std::transform(v.begin(), v.end(), v.begin(), do_something);

matrix.unaryExpr(do_something);

Answer 2

Eigen::VectorX*仅继承自Eigen::EigenBase

template<typename Scalar>
void do_something(Eigen::Matrix<Scalar, Eigen::Dynamic, 1> &evec) { /*...*/}

Answer 3

您可以使用std::enable_if仅为算术类型启用完全通用的版本：

template <typename T>
typename std::enable_if<std::is_arithmetic<T>::value,void>::type
do_something(T &var) { ... }