diff options
| -rw-r--r-- | src/Data/Array/Nested/Internal/Shaped.hs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/Data/Array/Nested/Internal/Shaped.hs b/src/Data/Array/Nested/Internal/Shaped.hs index ea979fa..d013959 100644 --- a/src/Data/Array/Nested/Internal/Shaped.hs +++ b/src/Data/Array/Nested/Internal/Shaped.hs @@ -399,7 +399,7 @@ sslice i n@SNat arr = srev1 :: Elt a => Shaped (n : sh) a -> Shaped (n : sh) a srev1 arr = slift (sshape arr) (\_ -> X.rev1) arr -sreshape :: Elt a => ShS sh' -> Shaped sh a -> Shaped sh' a +sreshape :: (Elt a, Product sh ~ Product sh') => ShS sh' -> Shaped sh a -> Shaped sh' a sreshape sh' (Shaped arr) = Shaped (mreshape (shCvtSX sh') arr) sflatten :: Elt a => Shaped sh a -> Shaped '[Product sh] a |