apaam/mlpack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update isrlu_impl.hpp

Browse Source
This commit is contained in:
abh2k
2021-02-15 10:24:01 +05:30
committed by GitHub
parent efcba77603
commit 0fe7756794
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/methods/ann/layer/isrlu_impl.hpp
+2 -2
View File
@@ -33,7 +33,7 @@ void ISRLU<InputDataType, OutputDataType>::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<InputDataType, OutputDataType>::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);
}
}
}