Eigen中系数自定义函数

Question

我有一个do_magic方法，它需要一个双精度值并将其加42。我想将这种方法应用于Eigen::Matrix或Eigen::Array的每个系数（也就是说，如果只有两种类型中的一种，我才会介意）。

这可能吗？

像这样：

Eigen::MatrixXd m(2, 2);  
m << 1,2,1,2;  
m.applyCoefficientWise(do_magic); 
// m is now 43, 44, 43, 44 

Answer 1

您可以使用unaryExpr，虽然这会返回一个新的视图到基质上，而不是让你修改要素到位。

复制的例子出来的文件：

double ramp(double x) 
{ 
    if (x > 0) 
    return x; 
    else 
    return 0; 
} 
int main(int, char**) 
{ 
    Matrix4d m1 = Matrix4d::Random(); 
    cout << m1 << endl << "becomes: " << endl << m1.unaryExpr(ptr_fun(ramp)) << endl; 
    return 0; 
}