Add the broadcast operator.

2025-06-20 04:00:28 +00:00 · 2023-06-24 19:16:03 +01:00
parent 96c098b6cd
commit 6b2cd9c51c
4 changed files with 44 additions and 0 deletions
--- a/src/op.rs
+++ b/src/op.rs
@ -22,6 +22,7 @@ pub(crate) enum Op {
        add: f64,
    },
    ToDType(Tensor),
+    Broadcast(Tensor),
    Exp(Tensor),
    Log(Tensor),
    Sin(Tensor),