From 0fe77567948df2aaa1ce711003c2fcdefc49d8cc Mon Sep 17 00:00:00 2001 From: abh2k <41710346+abh2k@users.noreply.github.com> Date: Mon, 15 Feb 2021 10:24:01 +0530 Subject: [PATCH] Update isrlu_impl.hpp --- src/mlpack/methods/ann/layer/isrlu_impl.hpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/mlpack/methods/ann/layer/isrlu_impl.hpp b/src/mlpack/methods/ann/layer/isrlu_impl.hpp index b69ffbf8bb..8ee468849f 100644 --- a/src/mlpack/methods/ann/layer/isrlu_impl.hpp +++ b/src/mlpack/methods/ann/layer/isrlu_impl.hpp @@ -33,7 +33,7 @@ void ISRLU::Forward( for (size_t i = 0; i < input.n_elem; ++i) { output(i) = (input(i) >= 0) ? input(i) : input(i) % - (1 / std::sqrt(1 + alpha * (input(i) % input(i)))); + (1 / arma::sqrt(1 + alpha * (input(i) % input(i)))); } if (!deterministic) @@ -42,7 +42,7 @@ void ISRLU::Forward( for (size_t i = 0; i < input.n_elem; ++i) { derivative(i) = (input(i) >= 0) ? 1 : - std::pow(output(i), 3) / std::pow(input(i), 3); + arma::pow(output(i), 3) / arma::pow(input(i), 3); } } }