Adding embedding op (not generic gather, no select).

2025-06-20 12:06:35 +00:00 · 2023-06-22 22:47:35 +02:00
parent 4ffdeb4e23
commit 5e54f37fe1
8 changed files with 154 additions and 0 deletions
--- a/src/op.rs
+++ b/src/op.rs
@ -7,6 +7,7 @@ pub(crate) enum Op {
    Sub(Tensor, Tensor),
    Div(Tensor, Tensor),
    Matmul(Tensor, Tensor),
+    Embedding(Tensor, Tensor),

    Cat(Vec<Tensor>, usize),