diff --git a/src/mlpack/methods/ann/layer/elu.hpp b/src/mlpack/methods/ann/layer/elu.hpp index 5e273a6dc1..a4b232377c 100644 --- a/src/mlpack/methods/ann/layer/elu.hpp +++ b/src/mlpack/methods/ann/layer/elu.hpp @@ -194,7 +194,7 @@ class ELU template void Fn(const arma::Mat& x, arma::Mat& y) { - y.set_size(size(x)); + y.set_size(arma::size(x)); for (size_t i = 0; i < x.n_elem; i++) {