Eigen3 replicate()用于矩阵向量cwiseProduct操作

时间:2014-08-25 19:08:20

标签: c++ matrix-multiplication eigen eigen3

我有以下代码:

Eigen::MatrixXf aMatrix( 3, 5 );
aMatrix <<
1, 0, 1, 0, 1,
0, 1, 0, 1, 0,
1, 1, 1, 1, 1;

Eigen::VectorXf aVector( 5 );
aVector << 3, 4, 5, 6, 7;

cout << aMatrix.cwiseProduct( aVector.replicate( 1, aMatrix.rows() ).transpose() ) << endl;

输出:

3 0 5 0 7
0 4 0 6 0
3 4 5 6 7

有没有比使用replicate()调用更有效的方法来实现这一目标?

2 个答案:

答案 0 :(得分:4)

解决(在How can I apply bsxfun like functionality at Eigen?的帮助下)

这些是等价的:

aMatrix.cwiseProduct( aVector.replicate( 1, aMatrix.rows() ).transpose() )
aMatrix.array().rowwise() * aVector.array().transpose()

答案 1 :(得分:2)

我不确定这是否更有效,但是后期乘以对角矩阵是另一种选择。

aMatrix * aVector.asDiagonal();

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

int main()
{

  Eigen::MatrixXf aMatrix( 3, 5 );
  aMatrix <<
    1, 0, 1, 0, 1,
    0, 1, 0, 1, 0,
    1, 1, 1, 1, 1;

  Eigen::VectorXf aVector( 5 );
  aVector << 3, 4, 5, 6, 7;

  std::cout << aMatrix * aVector.asDiagonal() << std::endl;

  return 0;
}