def emb_bwd

bendtorch-gpu@1.3.0.0 · main.bend:1337

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

1337def emb_bwd(+n: Nat, +v: Nat, +d: Nat, e: TA.Mat<v, d>, ids: List<&2, Nat>, dy: TA.Mat<n, d>, g: TA.Mat<v, d>) -> T.Bwd<TA.Mat<v, d>, List<&2, Nat>>:
1338  match dy g:
1339    case TA.Mat{da} TA.Mat{ga}: emb_b2(v, d, e, da, ga, ids_arr(n, ids))

Uses (6)

Used by (1)

bendtorch-gpu@1.3.0.0/main:Embedding