如何“移动”Eigen :: VectorXd s

时间:2017-02-24 15:46:02

标签: c++ c++11 move eigen

我最近一篇文章中的一位评论者告诉我,我需要更好地利用c ++ 11移动语义来处理我的代码中的瓶颈。以下是需要修复的简化版本。

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

void makeCopy(std::vector<Eigen::VectorXd> &oldV){
    int n = oldV.size();
    std::vector<Eigen::VectorXd> mandatoryCopy;
    mandatoryCopy.resize(n);

    for(int i = 0; i < n; i++){
        mandatoryCopy[i] = oldV[i];
    }

    // swap the two
    oldV = mandatoryCopy;
}

int main(int argc, char **argv)
{
    // starting vector
    int len(1000);
    Eigen::VectorXd placeHolder(50);
    std::vector<Eigen::VectorXd> v(len, placeHolder);

    // copy it a bunch of times
    for(int iter = 0; iter < 1000; ++iter){
        std::cout << "iter: " << iter << "\n";
        makeCopy(v);
    }

    return 0;
}

问题:在makeCopy的for循环中,oldV[i]是左值,所以我怎样才能做mandatoryCopy[i]&& = oldV[i]之类的事情?这是主要的瓶颈,对吧?我在想mandatoryCopy[i]&& = std::move(oldV[i])之类的东西,但这显然不会编译。

修改

根据@ vsoftco的建议,我试过

std::vector<Eigen::VectorXd> makeCopy2(std::vector<Eigen::VectorXd> oldV){

    int n = oldV.size();
    std::vector<Eigen::VectorXd> mandatoryCopy;
    mandatoryCopy.resize(n);

    for(int i = 0; i < n; i++){
        mandatoryCopy[i] = oldV[i];
    }
    return mandatoryCopy;
}

但我发现它慢了。 @vsoftco和@ggael都提到返回修改后的复制参数会更快,而不是再次复制,我同意,但我怀疑这对我的实际代码是可行的。我可以稍后再问这个问题,但这将是一个单独的问题/主题。

1 个答案:

答案 0 :(得分:2)

你没有看到正确的路线。如果一个副本是强制性的,那么你就无法摆脱它。尽管如此,最好避免使用for循环:

std::vector<Eigen::VectorXd> mandatoryCopy = oldV;

另一方面,您可以通过将oldV=mandatoryCopy替换为:

来省略第二个副本
std::swap(oldV,mandatoryCopy);

将执行便宜的指针交换。你得到:

void makeCopy(std::vector<Eigen::VectorXd> &oldV){
  std::vector<Eigen::VectorXd> V = oldV;
  // do something with V
  std::swap(oldV,V);
}

对于功能样式,在第二个示例中,您必须直接使用已经是副本的参数:

std::vector<Eigen::VectorXd> makeCopy2(std::vector<Eigen::VectorXd> V){
  // do something with V
  return V;
}

并使用v = makeCopy2(v);调用它。

不要忘记使用-std=c++11进行编译以获取移动语义副本。

最后,将vector<VectorXd>打包在MatrixXd内可能会更好。这将显着减少内存分配的数量:

void makeCopy3(MatrixXd &oldV){
  int n = oldV.cols();
  MatrixXd V = oldV;
  for(int i = 0; i < n; i++){
    V.col(i) *= 0.99;
  }
  oldV.swap(V); // or oldV = std::move(V); with c++11 enabled
}