Implement reshape

1 files changed, 6 insertions, 0 deletions

diff --git a/src/Analysis/Identity.hs b/src/Analysis/Identity.hs
index 9dc8811..b3a6664 100644
--- a/src/Analysis/Identity.hs
+++ b/src/Analysis/Identity.hs
@@ -244,6 +244,12 @@ idana env expr = case expr of
     res <- VIArr <$> genId <*> pure sh
     pure (res, EMinimum1Inner res e1')
 
+  EReshape _ dim e1 e2 -> do
+    (v1, e1') <- idana env e1
+    (_, e2') <- idana env e2
+    res <- VIArr <$> genId <*> shidsToVec dim v1
+    pure (res, EReshape res dim e1' e2')
+
   EFold1InnerD1 _ cm e1 e2 e3 -> do
     let t1 = typeOf e2
     x1 <- genIds t1