~/jendhub

def emb_add

bendtorch@1.3.0.0 · main.bend:2021

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

2021# g[gi + j] = g[gi + j] + dy[di + j], j = 0..k
2022def emb_add(k: Nat, +gi: U32, +di: U32, rg: Array<F32> & F32, rd: Array<F32> & F32) -> A2:
2023  match k rg rd:
2024    case 0n Tuple{g, +a} Tuple{dy, +b}:
2025      A2{Array.set(F32, g, gi, (a + b : F32)), dy}
2026    case 1n+q Tuple{g, +a} Tuple{dy, +b}:
2027      emb_add(q, (gi + 1 : U32), (di + 1 : U32), Array.get(F32, Array.set(F32, g, gi, (a + b : F32)), (gi + 1 : U32)), Array.get(F32, dy, (di + 1 : U32)))

Uses (11)

Used by (1)