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/Array.hs | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'src/Array.hs')

diff --git a/src/Array.hs b/src/Array.hs
index 707dce2..6ceb9fe 100644
--- a/src/Array.hs
+++ b/src/Array.hs
@@ -91,6 +91,11 @@ arrayFromList sh l = Array sh (V.fromListN (shapeSize sh) l)
 arrayToList :: Array n t -> [t]
 arrayToList (Array _ v) = V.toList v
 
+arrayReshape :: Shape n -> Array m t -> Array n t
+arrayReshape sh (Array sh' v)
+  | shapeSize sh == shapeSize sh' = Array sh v
+  | otherwise = error $ "arrayReshape: different shape size than original (" ++ show sh' ++ " -> " ++ show sh ++ ")"
+
 arrayUnit :: t -> Array Z t
 arrayUnit x = Array ShNil (V.singleton x)
 
-- 
cgit v1.2.3-70-g09d2