BendTensors.Flat.softmax_rows: def of bend-tensors@0.0.0.2
Import: import bend-tensors@0.0.0.2/bend_tensors.bend as BendTensors
Lines 1978-1981 of bend-tensors@0.0.0.2/bend_tensors.bend: https://jend.leeeet.dev/read.txt?path=bend-tensors@0.0.0.2/bend_tensors.bend&start_line=1978&end_line=1981

import Base

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

Uses (4):
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Nat
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=U32.from_nat
  https://jend.leeeet.dev/definition.txt?path=bend-tensors@0.0.0.2/bend_tensors.bend&name=Flat
  https://jend.leeeet.dev/definition.txt?path=bend-tensors@0.0.0.2/bend_tensors.bend&name=Flat.softmax.go