def Flat.zip
import bend-tensors@0.0.0.2/bend_tensors.bend as BendTensorsimport Base
1860# f over the pairs of cells; both operands are consumed
1861def Flat.zip(~f: F32 -> F32 -> F32, +rows: Nat, +cols: Nat, m: Flat<rows, cols>, n: Flat<rows, cols>) -> Flat<rows, cols>:
1862 FlatS{a} = m
1863 FlatS{b} = n
1864 Flat.zip.fin(rows, cols, Flat.zip.go(~f, Flat.count(rows, cols), 0, (a, b)))