c-特征:沿一维复制项目,而无用分配
作者:互联网
我有一些矢量vec,我想通过沿矢量维复制值来获得新的“表达式” vec2
Eigen::VectorXf vec(5);
vec << 1, 2, 3, 4, 5;
const auto vec2 = vec.someAwesomeEigenMagic<3>();
//vec2 should contains (1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4, 5, 5, 5)^T
//Not (1, 2, 3, 4, 5, 1, 2, 3, 4, 5, 1, 2, 3, 4, 5)^T
当然,我可以手动创建这种矢量,也可以使用Eigen :: Map使用复制矢量化创建矢量:
MatrixXf tmp = vec.replicate(1, 3).transpose();
const Map<VectorXf> vec2(tmp.data(), vec.rows() * 3, 1);
但是我希望vec2成为某种没有分配的“本征模板表达式”(vec可能很大,我会经常调用此例程)并立即计算值. (vec包含每个顶点权重,我想将其用于加权最小二乘法)
我考虑过用矢量的kronecker产品把戏,但是我不确定它是否针对逐个产品进行了优化.另外我更喜欢避免不受支持的模块
PS对不起,我的英语
解决方法:
使用devel分支,您可以使用LinSpaced生成索引序列,然后对输入向量进行索引:
#include <iostream>
#include <Eigen/Dense>
using namespace Eigen;
using namespace std;
int main()
{
VectorXf vec(5);
vec << 1, 2, 3, 4, 5;
auto vecrep = vec(ArrayXi::LinSpaced(5*3,0,4));
cout << vecrep.transpose() << endl;
}
然后可以在c 14中将关键行包装在返回auto的自由函数中:
template<typename XprType>
auto magic_rep(const XprType &xpr, Index K) {
return xpr(Eigen::ArrayXi::LinSpaced(xpr.size()*K,0,xpr.size()-1));
}
主要:
cout << magic_rep(vec,3).transpose() << endl;
标签:c,matrix,eigen,eigen3 来源: https://codeday.me/bug/20191011/1891366.html