代码之家 › 专栏 › 技术社区 › Daiver

特征:沿一维复制项,而不进行无用的分配

eigen3 eigen matrix c++

Daiver · 技术社区 · 7 年前

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

MatrixXf tmp = vec.replicate(1, 3).transpose();
const Map<VectorXf> vec2(tmp.data(), vec.rows() * 3, 1);

vec2 成为某种“特征模板表达式”,而不需要联合(vec可能相当大,我会经常调用此例程)和立即计算值。(vec包含逐顶点权重,我想将其用于加权最小二乘法)

我考虑过使用1的向量的克罗内克乘积技巧,但我不确定它是否针对逐个乘积进行了优化。此外,我更喜欢避免不支持的模块

1 回复 | 直到 7 年前

ggael 7 年前

使用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;
}

然后,您可以将关键行包装在一个自由函数中 auto ,在c++14中:

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;

推荐文章

avocado · 特征稀疏矩阵的性能调整

7 年前

Daiver · 特征:沿一维复制项,而不进行无用的分配

7 年前

zer0hedge · RcppEigen。h: 安装期间没有此类文件或目录。。。Centos 7上的RcppEigen

7 年前

Daiver · 特征:反转矩形矩阵(最小范数解)

7 年前

German Capuano · 以特征类型为参数的函数的返回类型不明确

7 年前

Daiver · 在没有内存开销的情况下设置特征::稀疏矩阵的稀疏模式

7 年前

Olivier · 使用MKL编译时,Eigen C++运行速度较慢

8 年前

Aditya Kashi · 使用valuePtr()等为特征3稀疏矩阵分配空间

8 年前