From 677a6d84871b77aabca96d782858e412041112ea Mon Sep 17 00:00:00 2001
From: Adam Kropp <30504149+akropp@users.noreply.github.com>
Date: Wed, 15 Nov 2023 10:17:02 -0500
Subject: [PATCH] input dimension size calc was wrong in Reset()

---
 src/mlpack/methods/ann/ffn_impl.hpp | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/mlpack/methods/ann/ffn_impl.hpp b/src/mlpack/methods/ann/ffn_impl.hpp
index 557c4a3fdc..2deda4090d 100644
--- a/src/mlpack/methods/ann/ffn_impl.hpp
+++ b/src/mlpack/methods/ann/ffn_impl.hpp
@@ -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
   {