From ed2427d5f7d39ea2330a8da1693d069f9a0eb81f Mon Sep 17 00:00:00 2001
From: Michael Marshall <43034299+mmphys@users.noreply.github.com>
Date: Sat, 6 Apr 2019 15:37:53 +0100
Subject: [PATCH] Make sure Grid::Serializable can write Eigen Tensors to
 output streams. NB: 1) The Eigen package defines operator<< for Eigen
 tensors, but this format is different, hence Grid::Serializable::WriteMember
 2) For simplification, the contents are written in memory order. I.e.
 Different results will be obtained depending on whether the tensor is row- or
 column-major

---
 Grid/serialisation/BaseIO.h | 22 +++++++++++++++++++++-
 1 file changed, 21 insertions(+), 1 deletion(-)

diff --git a/Grid/serialisation/BaseIO.h b/Grid/serialisation/BaseIO.h
index f11dcde5..e9d5bedf 100644
--- a/Grid/serialisation/BaseIO.h
+++ b/Grid/serialisation/BaseIO.h
@@ -583,7 +583,27 @@ namespace Grid {
     template <typename T>
     static inline typename std::enable_if<EigenIO::is_tensor<T>::value, void>::type
     WriteMember(std::ostream &os, const T &object) {
-      os << "Eigen::Tensor";
+      using Index = typename T::Index;
+      const Index NumElements{object.size()};
+      assert( NumElements > 0 );
+      Index count = 1;
+      os << "T<";
+      for( int i = 0; i < T::NumIndices; i++ ) {
+        Index dim = object.dimension(i);
+        count *= dim;
+        if( i )
+          os << ",";
+        os << dim;
+      }
+      assert( count == NumElements && "Number of elements doesn't match tensor dimensions" );
+      os << ">{";
+      const typename T::Scalar * p = object.data();
+      for( Index i = 0; i < count; i++ ) {
+        if( i )
+          os << ",";
+        os << *p++;
+      }
+      os << "}";
     }
   };