custom_layer.hpp
Go to the documentation of this file.
1 
13 #ifndef MLPACK_TESTS_CUSTOM_LAYER_HPP
14 #define MLPACK_TESTS_CUSTOM_LAYER_HPP
15 
16 #include <mlpack/prereqs.hpp>
18 
19 
20 namespace mlpack {
21 namespace ann {
25  template <
26  class ActivationFunction = LogisticFunction,
27  typename InputDataType = arma::mat,
28  typename OutputDataType = arma::mat
29  >
30  using CustomLayer = BaseLayer<
31  ActivationFunction, InputDataType, OutputDataType>;
32 
33 } // namespace ann
34 } // namespace mlpack
35 
36 #endif // MLPACK_TESTS_CUSTOM_LAYER_HPP
BaseLayer< ActivationFunction, InputDataType, OutputDataType > CustomLayer
Standard Sigmoid layer.
.hpp
Definition: add_to_po.hpp:21
The core includes that mlpack expects; standard C++ includes and Armadillo.