diff --git a/src/tensor.h b/src/tensor.h
index 4b1e186f..fdae329a 100644
--- a/src/tensor.h
+++ b/src/tensor.h
@@ -4,6 +4,7 @@
 #include <cublas_v2.h>
 #include <thrust/device_vector.h>
 #include <thrust/functional.h>
+#include <numeric>
 
 #include "definitions.h"
 #include "exception.h"