def bgroup_full
import bendtorch@1.3.0.0/main.bend as Bendtorchimport Base
3043def bgroup_full(+dm: Dm, +j: Nat, s: BG) -> BG:
3044 match dm s:
3045 case Dm{+t, +g, +hd, +w, +kvw, +sc} BG{q, k, v, d, dq, dk, dv}:
3046 bg_fin(t, hd, kvw, j, dk, dv, bheads_full(g, Nat.mul(j, g), dm, BS{q, k, v, d, dq, zeros_a(Nat.mul(t, hd)), zeros_a(Nat.mul(t, hd))}))