VTEN-11-Fix-get_size-bug (#14)

VORTICITY-INC · Nov 4, 2024 · 35e19dd · 35e19dd
1 parent de7bf0a
commit 35e19dd
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/lib/core/tensor.hpp b/lib/core/tensor.hpp
@@ -29,7 +29,9 @@ using Shape = std::array<size_t, N>;
  */
 template <size_t N>
 size_t get_size(const Shape<N>& shape) {
-    return std::accumulate(shape.begin(), shape.end(), 1, std::multiplies<size_t>());
+    size_t size = 1;
+    for (size_t i = 0; i < N; ++i) size *= shape[i];
+    return size;
 }
 
 /**