[mlpack-git] master: Generalize the linear forward method. (57b7eb7)
gitdub at big.cc.gt.atl.ga.us
gitdub at big.cc.gt.atl.ga.us
Thu Dec 31 12:48:37 EST 2015
Repository : https://github.com/mlpack/mlpack
On branch : master
Link : https://github.com/mlpack/mlpack/compare/29c0405173902f366d241e19dabfe3d679f8bea8...6945c11b8a626229ec6577ba72c88a74c2964e80
>---------------------------------------------------------------
commit 57b7eb71d4d8755ed2da4cd83c7fa606778c8a9e
Author: marcus <marcus.edel at fu-berlin.de>
Date: Mon Nov 30 21:53:09 2015 +0100
Generalize the linear forward method.
>---------------------------------------------------------------
57b7eb71d4d8755ed2da4cd83c7fa606778c8a9e
src/mlpack/methods/ann/layer/linear_layer.hpp | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/mlpack/methods/ann/layer/linear_layer.hpp b/src/mlpack/methods/ann/layer/linear_layer.hpp
index a556862..aebe173 100644
--- a/src/mlpack/methods/ann/layer/linear_layer.hpp
+++ b/src/mlpack/methods/ann/layer/linear_layer.hpp
@@ -98,8 +98,8 @@ class LinearLayer
* @param input Input data used for evaluating the specified function.
* @param output Resulting output activation.
*/
- template<typename eT>
- void Forward(const arma::Mat<eT>& input, arma::Mat<eT>& output)
+ template<typename InputType, typename OutputType>
+ void Forward(const InputType& input, OutputType& output)
{
output = weights * input;
}
More information about the mlpack-git
mailing list