Add transposition around arbitrary axis.

2025-06-20 04:00:28 +00:00 · 2023-06-23 08:51:13 +01:00
parent 27d428af1a
commit 3f79d81b6f
2 changed files with 25 additions and 11 deletions
--- a/src/op.rs
+++ b/src/op.rs
@ -17,7 +17,7 @@ pub(crate) enum Op {
    Neg(Tensor),
    Sqr(Tensor),
    Sqrt(Tensor),
-    Transpose(Tensor),
+    Transpose(Tensor, usize, usize),
    // TODO: Support for custom ops.
 }