def Flat.softmax_rows

bend-tensors@0.0.0.2 · bend_tensors.bend:1978

import Base

1978# softmax along each row
1979def Flat.softmax_rows(+rows: Nat, +cols: Nat, m: Flat<rows, cols>) -> Flat<rows, cols>:
1980  FlatS{a} = m
1981  FlatS{Flat.softmax.go(rows, cols, U32.from_nat(cols), 0, a)}