Use DenseIndex in the MeanReducer to avoid overflows when processing very large tensors.

2026-04-10 11:34:33 +08:00 · 2016-04-19 11:53:58 -07:00
parent 84543c8be2
commit 50968a0a3e
1 changed files with 2 additions and 2 deletions
--- a/unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h
+++ b/unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h
@@ -158,8 +158,8 @@ template <typename T> struct MeanReducer
  }

  protected:
-    int scalarCount_;
-    int packetCount_;
+    DenseIndex scalarCount_;
+    DenseIndex packetCount_;
 };

 template <typename T> struct MaxReducer