mlpack 3.4.2
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
20namespace mlpack {
21namespace ann {
25 template <
26 class ActivationFunction = LogisticFunction,
27 typename InputDataType = arma::mat,
28 typename OutputDataType = arma::mat
29 >
31 ActivationFunction, InputDataType, OutputDataType>;
32
33} // namespace ann
34} // namespace mlpack
35
36#endif // MLPACK_TESTS_CUSTOM_LAYER_HPP
Implementation of the base layer.
Definition: base_layer.hpp:66
Linear algebra utility functions, generally performed on matrices or vectors.
Definition: cv.hpp:1
The core includes that mlpack expects; standard C++ includes and Armadillo.