From 4d456e4d34b1e4fb3725051d1b8a0c376b704692 Mon Sep 17 00:00:00 2001
From: Tom Smeding <tom@tomsmeding.com>
Date: Thu, 30 Oct 2025 15:56:35 +0100
Subject: Implement reshape

---
 src/Language.hs | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'src/Language.hs')

diff --git a/src/Language.hs b/src/Language.hs
index 4e6d604..d3c38d6 100644
--- a/src/Language.hs
+++ b/src/Language.hs
@@ -130,6 +130,9 @@ maximum1i e = NEMaximum1Inner e
 minimum1i :: ScalIsNumeric t ~ True => NExpr env (TArr (S n) (TScal t)) -> NExpr env (TArr n (TScal t))
 minimum1i e = NEMinimum1Inner e
 
+reshape :: SNat n -> NExpr env (Tup (Replicate n TIx)) -> NExpr env (TArr m t) -> NExpr env (TArr n t)
+reshape = NEReshape
+
 const_ :: KnownScalTy t => ScalRep t -> NExpr env (TScal t)
 const_ x =
   let ty = knownScalTy
-- 
cgit v1.2.3-70-g09d2