Add the backward pass for transpose.

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