def emb_f

bendtorch-gpu@1.3.0.0 · main.bend:1325

import bendtorch-gpu@1.3.0.0/main.bend as BendtorchGpu

import Base
import bend-ml-tensor-array@0.1.18.0/main.bend as TA
import bendtorch@1.3.0.0/main.bend as T

1325def emb_f(+n: Nat, +v: Nat, +d: Nat, +ids: List<&2, Nat>, e: TA.Mat<v, d>) -> T.Fwd<TA.Mat<v, d>, TA.Mat<n, d>, List<&2, Nat>>:
1326  match e:
1327    case TA.Mat{ea}: emb_f2(n, v, d, ids, ea, ids_arr(n, ids))

Uses (6)

Used by (1)

bendtorch-gpu@1.3.0.0/main:Embedding