compat/mat: add elementwise mult

1 files changed, 11 insertions, 0 deletions

diff --git a/compat/simple-mat.hpp b/compat/simple-mat.hpp
index 5bc26a62..a806448b 100644
--- a/compat/simple-mat.hpp
+++ b/compat/simple-mat.hpp
@@ -199,6 +199,17 @@ public:
         return ret;
     }
 
+    constexpr Mat<num, h_, w_> mult_elementwise(const Mat<num, h_, w_>& other) const&
+    {
+        Mat<num, h_, w_> ret;
+
+        for (unsigned j = 0; j < h_; j++)
+            for (unsigned i = 0; i < w_; i++)
+                ret(j, i) = data[j][i] * other.data[j][i];
+
+        return ret;
+    }
+
     constexpr inline num operator()(int j, int i) const& { return data[j][i]; }
     constexpr inline num& operator()(int j, int i) & { return data[j][i]; }