input dimension size calc was wrong in Reset()

2023-11-15 10:17:02 -05:00
parent d90282abf5
commit 677a6d8487
1 changed files with 5 additions and 3 deletions
@@ -252,9 +252,11 @@ void FFN<
  }
  else if (inputDimensions.size() > 0)
  {
-    const size_t inputDims = std::accumulate(inputDimensions.begin(),
-        inputDimensions.end(), 0);
-    CheckNetwork("FFN::Reset()", inputDims, true, false);
+    size_t inputDim = inputDimensions[0];
+    for (size_t i=1; i<inputDimensions.size(); i++) {
+      inputDim *= inputDimensions[i];
+    }
+    CheckNetwork("FFN::Reset()", inputDim, true, false);
  }
  else
  {