mlpack
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>
17 #include <mlpack/methods/ann/layer/layer.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
mlpack::ann::CustomLayer
BaseLayer< ActivationFunction, InputDataType, OutputDataType > CustomLayer
Standard Sigmoid layer.
Definition: custom_layer.hpp:31
mlpack
Linear algebra utility functions, generally performed on matrices or vectors.
Definition: add_to_cli11.hpp:21
prereqs.hpp
The core includes that mlpack expects; standard C++ includes and Armadillo.