~/jendhub

def smf_rows

bendtorch@1.3.0.0 · main.bend:1397

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

1397# rows i.. of x (c wide, left more after i), each softmaxed in place (psm)
1398def smf_rows(left: Nat, +i: U32, +c: Nat, x: Array<F32>) -> Array<F32>:
1399  match left:
1400    case 0n:
1401      psm(c, (i * U32.from_nat(c) : U32), x)
1402    case 1n+q:
1403      smf_rows(q, (i + 1 : U32), c, psm(c, (i * U32.from_nat(c) : U32), x))

Uses (8)

Used by (1)