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

minor change

Browse Source
This commit is contained in:
Abhinav Anand
2021-02-06 02:49:15 +05:30
parent a88994e48b
commit c1c0cfbc18
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/lp_pooling.hpp
+2 -2
View File
@@ -175,7 +175,7 @@ class LpPooling
arma::span(rowidx, rowidx + kernelWidth - 1 - offset),
arma::span(colidx, colidx + kernelHeight - 1 - offset));
output(i, j) = arma::pow(arma::accu(arma::pow(subInput, norm_type)), 1.0/norm_type);
output(i, j) = cmath::pow(arma::accu(arma::pow(subInput, norm_type)), 1.0/norm_type);
}
}
}
@@ -201,7 +201,7 @@ class LpPooling
{
const arma::Mat<eT>& inputArea = input(arma::span(i, i + rStep - 1),
arma::span(j, j + cStep - 1));
size_t sum = arma::pow(arma::accu(arma::pow(inputArea, norm_type)), (norm_type-1) / norm_type);
size_t sum = cmath::pow(arma::accu(arma::pow(inputArea, norm_type)), (norm_type-1) / norm_type);
unpooledError = arma::Mat<eT>(inputArea.n_rows, inputArea.n_cols);
unpooledError.fill(error(i / rStep, j / cStep));
unpooledError %= arma::pow(inputArea, norm_type - 1);