revised l2pool2d

webmachinelearning · Jan 12, 2024 · 46ce01c · 46ce01c
1 parent de57d69
commit 46ce01c
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/src/reduce.js b/src/reduce.js
@@ -145,15 +145,12 @@ export function reduceL1(input, options = {}) {
 
 /* The l2 reducer */
 export function l2Reducer(previousValue, currentValue, currentIndex, array) {
-  if (currentIndex== 1) {
-    const sumOfSquares = previousValue*previousValue + currentValue * currentValue;
+  if (currentIndex == 1) {
+    const sumOfSquares = previousValue * previousValue + currentValue * currentValue;
     return sumOfSquares;
-  } else if (currentIndex === array.length - 1) {
-    const sumOfSquares = previousValue + currentValue * currentValue;
-    return Math.sqrt(sumOfSquares);
   } else {
     const sumOfSquares = previousValue + currentValue * currentValue;
-    return sumOfSquares;
+    return (currentIndex === array.length - 1) ? Math.sqrt(sumOfSquares) :sumOfSquares;
   }
 }