c++ - 将 Armadillo 函数传递给用户定义的函数

Question

是否有可能以优雅的方式将 Armadillo 函数传递给任意函数？具体来说，我希望能够将 arma::tanh 传递给我的神经网络，以便它计算网络输出，即类似于此的内容:

arma::mat y = network.ForwardPropagate(x, arma::tanh);

Answer 1

通过 .transform() 在 Armadillo 矩阵上使用您选择的函数可能更容易成员函数。

使用 C++11 的示例:

mat A(100,100,fill::randu);

A.transform( [](double val) { return tanh(val); } );