将函数应用于Eigen Matrix中的所有元素，无需循环

Question

我有一个Eigen :: Matrix，我想生成一个新矩阵，其中所有元素都是通过对矩阵元素的一些函数调用生成的：

Matrix< Foo,2,2 > m = ...;
Matrix< int, 2, 2> new_m;

for each m[i][j]:
  new_m[i][j] = m[i][j].member_of_foo_returns_int()

我查看了Eigen :: unaryExpr，但元素发生了变化，返回必须相同。但是，我在第一个矩阵中有Foo对象，并在新矩阵中返回一个int。这可能没有香草循环吗？

Answer 1

您可以将lambda表达式传递给unaryExpr，如下所示：

Eigen::Matrix<int,2,2> new_m = m.unaryExpr(
    [](const Foo& x) {
        return x.member_of_foo_returns_int();
    });

如果你不能使用c ++ 11，你需要编写一个小辅助函数：

int func_wrapper(const Foo& x) {
    return x.member_of_foo_returns_int();
}

并使用std::ptr_fun传递：

Eigen::Matrix<int,2,2> new_m = m.unaryExpr(std::ptr_fun(func_wrapper));

对于调用成员函数，实际上已经实现了一个名为std::mem_fun_ref的良好辅助函数（这需要一个成员函数指针并返回unaryExpr接受的函子对象）：

Eigen::Matrix<int,2,2> new_m = m.unaryExpr(
    std::mem_fun_ref(&Foo::member_of_foo_returns_int));

所有这些变体都是类型安全的，即尝试将结果存储在非int中 - 矩阵将无法编译。