Add some unary ops.

2025-06-20 12:06:35 +00:00 · 2023-06-20 12:04:01 +01:00
parent b4d332cd1b
commit d9cb1917ce
4 changed files with 158 additions and 73 deletions
--- a/src/op.rs
+++ b/src/op.rs
@ -4,5 +4,7 @@ use crate::Tensor;
 pub(crate) enum Op {
    Add(Tensor, Tensor),
    Mul(Tensor, Tensor),
+    Sqr(Tensor),
+    Sqrt(Tensor),
    // TODO: Support for custom ops.
 }