Adapt CustomLayer.

This commit is contained in:
Ryan Curtin
2022-03-30 22:07:27 -04:00
parent 53d903d1c4
commit bcbeebcf27
+12 -11
View File
@@ -2,7 +2,7 @@
* @file tests/custom_layer.hpp
* @author Projyal Dev
*
* A simple custom layer mimicing SigmoidLayer for testing if custom
* A simple custom layer mimicking SigmoidLayer for testing if custom
* layers work.
*
* mlpack is free software; you may redistribute it and/or modify it under the
@@ -20,16 +20,17 @@
namespace mlpack {
namespace ann {
/**
* Standard Sigmoid layer.
*/
template <
class ActivationFunction = LogisticFunction,
typename InputDataType = arma::mat,
typename OutputDataType = arma::mat
>
using CustomLayer = BaseLayer<
ActivationFunction, InputDataType, OutputDataType>;
/**
* Standard Sigmoid layer.
*/
template <
class ActivationFunction = LogisticFunction,
typename InputDataType = arma::mat,
typename OutputDataType = arma::mat
>
using CustomLayer = BaseLayer<
ActivationFunction, InputDataType, OutputDataType>;
} // namespace ann
} // namespace mlpack