diff --git a/src/mlpack/methods/ann/layer/c_relu_impl.hpp b/src/mlpack/methods/ann/layer/c_relu_impl.hpp index e29ea3ff83..e839526976 100644 --- a/src/mlpack/methods/ann/layer/c_relu_impl.hpp +++ b/src/mlpack/methods/ann/layer/c_relu_impl.hpp @@ -30,7 +30,7 @@ void CReLU::Forward( const InputType&& input, OutputType&& output) { output = arma::join_cols(arma::max(input, 0.0 * input), arma::max( - (-1 * input), 0.0 * input)); + (-1 * input), 0.0 * input)); } template @@ -41,7 +41,7 @@ void CReLU::Backward( DataType temp; temp = gy % (input >= 0.0); g = temp.rows(0, (input.n_rows / 2 - 1)) - temp.rows(input.n_rows / 2, - (input.n_rows - 1)); + (input.n_rows - 1)); } template