Add the reshape method and operation (without grad for now).

2025-06-20 04:00:28 +00:00 · 2023-06-23 10:51:05 +01:00
parent c4c6167949
commit 79e4b29c2f
2 changed files with 30 additions and 1 deletions
--- a/src/op.rs
+++ b/src/op.rs
@ -17,7 +17,6 @@ pub(crate) enum Op {
        add: f64,
    },
    Neg(Tensor),
-    #[allow(dead_code)]
    Reshape(Tensor),
    Sqr(Tensor),
    Sqrt(Tensor),