def prow3

bendtorch@1.3.0.0 · main.bend:2415

import bendtorch@1.3.0.0/main.bend as Bendtorch

import Base

2415# the softmax over the n numbers pw wrote; the row's count is the one the softmax reports
2416def prow3(+off: U32, q: Array<F32>, k: Array<F32>, w: PW) -> PR:
2417  match w:
2418    case PW{p, +n}:
2419      pr_of(q, k, psmc(n, off, p))

Uses (7)

Used by (3)