def bgroup
import bendtorch@1.3.0.0/main.bend as Bendtorchimport Base
2707def bgroup(+dm: Dm, +j: Nat, s: BG) -> BG:
2708 match dm s:
2709 case Dm{+t, +g, +hd, +w, +kvw, +sc} BG{q, k, v, d, dq, dk, dv}:
2710 bg_fin(t, hd, kvw, j, dk, dv, bheads(g, Nat.mul(j, g), dm, BS{q, k, v, d, dq, zeros_a(Nat.mul(t, hd)), zeros_a(Nat.mul(t, hd))}))