Add stream operator for Index (#1116)

This commit adds stream operator for Index

diff --git a/contrib/nnc/libs/core/include/nnc/core/linalg/Index.h b/contrib/nnc/libs/core/include/nnc/core/linalg/Index.h
index 0e329e6..9394c99 100644 (file)
--- a/contrib/nnc/libs/core/include/nnc/core/linalg/Index.h
+++ b/contrib/nnc/libs/core/include/nnc/core/linalg/Index.h
@@ -4,6 +4,7 @@
 #include <initializer_list>
 #include <vector>
 #include <cstdint>
+#include <ostream>
 
 namespace nncc
 {
@@ -37,6 +38,8 @@ private:
   std::vector<uint32_t> _indices;
 };
 
+std::ostream &operator<<(std::ostream &s, const Index &sh);
+
 } // namespace data
 } // namespace core
 } // namespace contrib

diff --git a/contrib/nnc/libs/core/src/core/linalg/Index.cpp b/contrib/nnc/libs/core/src/core/linalg/Index.cpp
index 102006c..94f97b1 100644 (file)
--- a/contrib/nnc/libs/core/src/core/linalg/Index.cpp
+++ b/contrib/nnc/libs/core/src/core/linalg/Index.cpp
@@ -32,6 +32,20 @@ Index &Index::fill(uint32_t index)
 uint32_t &Index::at(uint32_t axis) { return _indices.at(axis); }
 uint32_t Index::at(uint32_t axis) const { return _indices.at(axis); }
 
+std::ostream &operator<<(std::ostream &s, const Index &sh)
+{
+  s << "[ ";
+  for (uint32_t i = 0; i < sh.rank(); ++i)
+  {
+    if (i != 0 )
+      s << ", ";
+    s << sh.at(i);
+  }
+  s << "]";
+
+  return s;
+}
+
 } // namespace data
 } // namespace core
 } // namespace contrib