toki/numidr
1
0
Fork 0
Code Issues Pull requests Activity

Generalize transpose to all arrays

Browse source
This commit is contained in:
Kiana Sheibani 2022-08-31 16:01:21 -04:00
parent ea824a901b
commit 92fc8f00c9
Signed by: toki
GPG key ID: 6CB106C25E86A9F7
2 changed files with 9 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

9
src/Data/NumIdr/Array/Array.idr
View file

@ -484,6 +484,15 @@ stack axis arrs = rewrite sym (lengthCorrect arrs) in
getAxisInd FZ (i :: is) = (i, is)
getAxisInd {s=_::_} (FS ax) (i :: is) = mapSnd (i::) (getAxisInd ax is)
export
transpose : Array s a -> Array (reverse s) a
transpose arr with (viewShape arr)
_ | Shape s = fromFunctionNB (reverse s) (\is => arr !# reverse is)
export
(.T) : Array s a -> Array (reverse s) a
(.T) = transpose
--------------------------------------------------------------------------------
-- Implementations