def Ew.zip

bendtorch@1.3.0.0 · main.bend:192

import Base
import bend-ml-tensor-array@0.1.18.0/main.bend as TA

192# (f(a, b), b), element by element
193def Ew.zip(~f: @+a: F32 -> @+b: F32 -> F32, +r: Nat, +c: Nat, a: TA.Mat<r, c>, b: TA.Mat<r, c>) -> TA.MPair<r, c>:
194  match a b:
195    case TA.Mat{x} TA.Mat{y}:
196      ew2_fin(r, c, ew2_l(~f, Nat.mul(r, c), 0, Array.get(F32, x, 0), Array.get(F32, y, 0)))