changes acc to suggestions

2022-07-26 17:09:53 +08:00
parent 432f8dd9d3
commit 9d1dd81c29
1 changed files with 3 additions and 2 deletions
@@ -308,8 +308,9 @@ void BatchNormType<MatType>::Backward(
  // Step 4: sum (dl / dxhat * -1 / stdInv) + variance *
  // sum (-2 * (x - mu)) / m.
  MatType normTemp = arma::sum(arma::sum((norm.each_slice() %
-      arma::repmat(-stdInv, inputSize, 1)) + (inputMean.each_slice() % arma::repmat(vars, inputSize, 1) * (-2.0) / m), 2), 0) /
-      m;
+      arma::repmat(-stdInv, inputSize, 1)) + 
+      (inputMean.each_slice() % arma::repmat(vars, inputSize, 1) * (-2.0) / m),
+      2), 0) / m;
  gTemp.each_slice() += arma::repmat(normTemp, inputSize, 1);
 }