huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-20 12:06:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #6 from LaurentMazare/add_embedding

Browse Source 
Adding embedding op (not generic gather, no select).
This commit is contained in:
Nicolas Patry
2023-06-23 13:49:13 +02:00
committed by GitHub
parent 52c503ba8f 2fb87edda5
commit 7c1625f6a5
8 changed files with 160 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/op.rs
View File

@ -7,6 +7,7 @@ pub(crate) enum Op {
Sub(Tensor, Tensor),
Div(Tensor, Tensor),
Matmul(Tensor, Tensor),
Embedding(Tensor, Tensor),
Cat(Vec<Tensor>, usize),