将函数应用于所有Eigen矩阵元素

Question

我有一个Eigen::MatrixXd,我想通过应用组件功能来修改其所有元素.例如:

MatrixXd m = ...;

for each m[i][j]:
  m[i][j] = exp(m[i][j]);

有没有办法实现这个结果？

Answer 1

是的,使用Eigen::MatrixBase<>::unaryExpr()成员函数.例:

#include <cmath>
#include <iostream>

#include <Eigen/Core>

double Exp(double x) // the functor we want to apply
{
    return std::exp(x);
}

int main()
{
    Eigen::MatrixXd m(2, 2);
    m << 0, 1, 2, 3;
    std::cout << m << std::endl << "becomes: ";
    std::cout << std::endl << m.unaryExpr(&Exp) << std::endl;
}

Answer 2

vsoftco的答案非常通用,适用于自定义功能.但是,对于许多常用功能,有一种更简单的方法.调整他的例子,我们可以使用arrays,它看起来像这样:

#include <iostream>
#include <Eigen/Core>

int main()
{
    Eigen::MatrixXd m(2, 2);
    m << 0, 1, 2, 3;
    std::cout << m << "\nbecomes:\n";
    std::cout << m.array().exp() << std::endl;
    return 0;
}

Answer 3

FWIW，在 C++11 及更高版本中，这也适用于 lambda 函数。

#include <cmath>
#include <iostream>

#include <Eigen/Core>

int main()
{
    Eigen::MatrixXd m(2, 2);
    m << 0, 1, 2, 3;
    std::cout << m << std::endl << " ->  " 
    std::cout << m.unaryExpr([](double x){return x + 1}) << std::endl;
}