def Flat.map
import bend-tensors@0.0.0.2/bend_tensors.bend as BendTensorsimport Base
1804def Flat.map(~f: F32 -> F32, +rows: Nat, +cols: Nat, m: Flat<rows, cols>) -> Flat<rows, cols>:
1805 FlatS{a} = m
1806 FlatS{Flat.map.go(~f, Flat.count(rows, cols), 0, a)}