custom_layer.hpp

Go to the documentation of this file.

 
#ifndef MLPACK_TESTS_CUSTOM_LAYER_HPP
#define MLPACK_TESTS_CUSTOM_LAYER_HPP
 
#include <mlpack/prereqs.hpp>
#include <mlpack/methods/ann/layer/layer.hpp>
 
 
namespace mlpack {
namespace ann {
  template <
      class ActivationFunction = LogisticFunction,
      typename InputDataType = arma::mat,
      typename OutputDataType = arma::mat
  >
  using CustomLayer = BaseLayer<
      ActivationFunction, InputDataType, OutputDataType>;
 
} // namespace ann
} // namespace mlpack
 
#endif // MLPACK_TESTS_CUSTOM_LAYER_HPP