From 85e31eaad0ee25d71d025bdf50a92d59b3b797bb Mon Sep 17 00:00:00 2001
From: Stanislaw Halik <sthalik@misaki.pl>
Date: Wed, 1 Nov 2017 13:50:05 +0100
Subject: compat/mat: add elementwise mult

---
 compat/simple-mat.hpp | 11 +++++++++++
 1 file changed, 11 insertions(+)

(limited to 'compat')

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]; }
 
-- 
cgit v1.2.3