[mlpack-git] master: Provide a few convenience typedefs. (7eece24)
gitdub at big.cc.gt.atl.ga.us
gitdub at big.cc.gt.atl.ga.us
Wed Sep 2 07:33:54 EDT 2015
Repository : https://github.com/mlpack/mlpack
On branch : master
Link : https://github.com/mlpack/mlpack/compare/86771a45890936ed536045420b0912acbe638b80...4c0c618df62d8bf1a7cc3cb9e3b806bc127044da
>---------------------------------------------------------------
commit 7eece24e36f8cab8faa0364957076f3c305c90fe
Author: Marcus Edel <marcus.edel at fu-berlin.de>
Date: Tue Sep 1 15:22:28 2015 +0200
Provide a few convenience typedefs.
>---------------------------------------------------------------
7eece24e36f8cab8faa0364957076f3c305c90fe
src/mlpack/methods/ann/layer/base_layer.hpp | 12 ++++++++++++
1 file changed, 12 insertions(+)
diff --git a/src/mlpack/methods/ann/layer/base_layer.hpp b/src/mlpack/methods/ann/layer/base_layer.hpp
index a4ad6ca..6cc3be9 100644
--- a/src/mlpack/methods/ann/layer/base_layer.hpp
+++ b/src/mlpack/methods/ann/layer/base_layer.hpp
@@ -23,6 +23,7 @@ namespace ann /** Artificial Neural Network. */ {
*
* - SigmoidLayer
* - IdentityLayer
+ * - BaseLayer2D
*
* @tparam ActivationFunction Activation function used for the embedding layer.
* @tparam InputDataType Type of the input data (arma::colvec, arma::mat,
@@ -164,6 +165,17 @@ template <
using IdentityLayer = BaseLayer<
ActivationFunction, InputDataType, OutputDataType>;
+/**
+ * Standard Base-Layer2D using the logistic activation function.
+ */
+template <
+ class ActivationFunction = LogisticFunction,
+ typename InputDataType = arma::cube,
+ typename OutputDataType = arma::cube
+>
+using BaseLayer2D = BaseLayer<
+ ActivationFunction, InputDataType, OutputDataType>;
+
}; // namespace ann
}; // namespace mlpack
More information about the mlpack-git
mailing list