add operator<<(std::ostream, Dual)

LoopModels · Feb 16, 2024 · 4c7c27a · 4c7c27a
1 parent faa6dab
commit 4c7c27a
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/include/Math/Dual.hpp b/include/Math/Dual.hpp
@@ -8,6 +8,7 @@
 #include <Math/StaticArrays.hpp>
 #include <Utilities/Invariant.hpp>
 #include <cstddef>
+#include <istream>
 #include <utility>
 
 namespace poly::math {
@@ -824,6 +825,14 @@ constexpr auto dval(Dual<T, N> &x) -> double & {
   return dval(x.value());
 }
 
+template <typename T, ptrdiff_t N>
+auto operator<<(std::ostream &os, const Dual<T, N> &x) -> std::ostream & {
+  os << "Dual<" << N << ">{" << x.value();
+  for (ptrdiff_t n = 0; n < N; ++n) os << ", " << x.gradient()[n];
+  os << "}";
+  return os;
+};
+
 class GradientResult {
   double x;
   MutPtrVector<double> grad;