def Flat.softmax.go

bend-tensors@0.0.0.2 · bend_tensors.bend:1971

import bend-tensors@0.0.0.2/bend_tensors.bend as BendTensors
import Base

1971def Flat.softmax.go(n: Nat, +cols: Nat, +colsu: U32, +base: U32, a: Array<F32>) -> Array<F32>:
1972  match n:
1973    case 0n:
1974      a
1975    case 1n+p:
1976      Flat.softmax.go(p, cols, colsu, (base + colsu : U32), Flat.softmax.row(cols, base, a))

Uses (6)

Used by (1)