~/jendhub

def emb_sc

bendtorch@1.3.0.0 · main.bend:2029

import bendtorch@1.3.0.0/main.bend as Bendtorch
import Base

2029# each dY row added into the gradient's row ids[i], in order, by index
2030def emb_sc(left: Nat, ids: List<&2, Nat>, +i: U32, +d: Nat, st: A2) -> A2:
2031  match left ids st:
2032    case 1n+q Con{+id, t} A2{g, dy}:
2033      emb_sc(q, t, (i + 1 : U32), d, emb_add(Nat.sub(d, 1n), (U32.from_nat(id) * U32.from_nat(d) : U32), (i * U32.from_nat(d) : U32), Array.get(F32, g, (U32.from_nat(id) * U32.from_nat(d) : U32)), Array.get(F32, dy, (i * U32.from_nat(d) : U32))))
2034    case _ _ st:
2035      st

Uses (11)

Used by (1)