Adapt CustomLayer for testing to have only one template parameter.

2022-04-02 15:37:55 -04:00
parent 9a9b6334a6
commit 232ce2a342
1 changed files with 2 additions and 4 deletions
@@ -26,11 +26,9 @@ namespace ann {
 */
 template <
    class ActivationFunction = LogisticFunction,
-    typename InputDataType = arma::mat,
-    typename OutputDataType = arma::mat
+    typename MatType = arma::mat
 >
-using CustomLayer = BaseLayer<
-    ActivationFunction, InputDataType, OutputDataType>;
+using CustomLayer = BaseLayer<ActivationFunction, MatType>;

 } // namespace ann
 } // namespace mlpack