def emb_bwd

bendtorch@1.3.0.0 · main.bend:2042

import bendtorch@1.3.0.0/main.bend as Bendtorch

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

2042def 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>) -> Bwd<TA.Mat<v, d>, List<&2, Nat>>:
2043  match dy g:
2044    case TA.Mat{da} TA.Mat{ga}:
2045      emb_b2(v, d, e, emb_sc(n, ids, 0, d, A2{ga, da}))

Uses (7)

Used by (2)