From 5fa2465ee659fc58571d05a4f20cbee4b1161514 Mon Sep 17 00:00:00 2001 From: manthan-r-sheth Date: Wed, 14 Mar 2018 21:28:07 +0530 Subject: [PATCH] Removed use of repmat --- src/mlpack/methods/ann/layer/batch_norm_impl.hpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/mlpack/methods/ann/layer/batch_norm_impl.hpp b/src/mlpack/methods/ann/layer/batch_norm_impl.hpp index 504e66ee84..d8f0e2c90f 100644 --- a/src/mlpack/methods/ann/layer/batch_norm_impl.hpp +++ b/src/mlpack/methods/ann/layer/batch_norm_impl.hpp @@ -89,8 +89,8 @@ void BatchNorm::Backward( variance = arma::var(input, 1, 1); arma::mat m = arma::sum(gy % (input.each_col() - mean), 1); - g = arma::repmat(m, 1, input.n_cols) % (mean - input.each_col()); - g.each_col() %= 1.0 / (variance + eps); + g = (mean - input.each_col()); + g.each_col() %= m; g += (gy.each_col() - arma::sum(gy, 1)); g += (input.n_cols - 1) * gy; g.each_col() %= ((1.0 / input.n_cols) * gamma);