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