我写了一个函数
joins the rows of two 2D arrays:
template <typename S, typename T>
Array<typename S::Scalar, Dynamic, Dynamic> join_rows(const ArrayBase<S> & A, const ArrayBase<T> & B) {
Array<typename S::Scalar, Dynamic, Dynamic> C (A.rows(), A.cols()+B.cols());
C << A, B;
return C;
}
我想写一个更通用的函数,可以连接两个以上的数组.
它应该能够与任何可迭代容器一起工作,例如. std :: list或std :: vector,所以我会使用模板模板paratemeter.
我可以很容易地用两个for循环来调整函数体,这不是问题,我只是在努力弄清楚这个函数的正确类型是什么.
(ps.我甚至不确定我的上面代码是否具有最佳类型,但它似乎可以完成这项工作)
最佳答案 我不确定如何声明任意数组的向量,但是你可以实现一个函数模板,它结合了一个或多个直接传递给它的参数.这通常通过递归调用自身,处理每个连续的参数来完成:
// end case (one argument): just forward the same array
template <typename T>
T&& join_rows(T&& A) {
return std::forward<T>(A);
}
// main function template: two or more arguments
template <typename S, typename T, typename... R>
Array<typename S::Scalar, Dynamic, Dynamic> join_rows(const ArrayBase<S>& A,
const ArrayBase<T>& B,
const ArrayBase<R>&... rest) {
Array<typename S::Scalar, Dynamic, Dynamic> C(A.rows(), A.cols()+B.cols());
C << A, B;
return join_rows(C, rest...); // call with the first two arguments combined
}
示例用于说明用法:
int main() {
Array<int, 1, 3> arr1 = {1, 2, 3};
Array<int, 1, 2> arr2 = {4, 5};
Array<int, 1, 4> arr3 = {9, 8, 7, 6};
cout << join_rows(arr1, arr2, arr3.reverse()) << endl; // 1 2 3 4 5 6 7 8 9
return 0;
}
如果要将单参数join_rows限制为仅接受Eigen :: Arrays,请为ArrayBase添加std :: enable_if检查< T>基类:
template <typename T>
std::enable_if_t<std::is_base_of<ArrayBase<std::decay_t<T>>,std::decay_t<T>>::value, T&&>
join_rows(T&& A) {
return std::forward<T>(A);
}
对于大型阵列,可能有更有效的方法来实现它.您可能会返回一个仅分配一个新Array对象的代理对象.