def fprobs

bendtorch@1.3.0.0 · main.bend:2469

import bendtorch@1.3.0.0/main.bend as Bendtorch

import Base

2469def fprobs(+m: Nat, +hd: Nat, +sc: F32, +qs: U32, +qc: U32, +ks: U32, +kc: U32, q: Array<F32>, k: Array<F32>) -> QK:
2470  fprows(m, 0n, m, hd, sc, qs, qc, ks, kc, fprow(0n, m, hd, sc, qs, qc, ks, kc, q, k, zeros_a(Nat.mul(1n+m, 1n+m))))

Uses (9)

Used by (2)