def Mat.sum2

bendtorch-fast@1.3.0.0 · main.bend:239

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

239# a + b, dropping b (T.Mat.sum2)
240def Mat.sum2(+r: Nat, +c: Nat, a: TA.Mat<r, c>, b: TA.Mat<r, c>) -> TA.Mat<r, c>:
241  mpair_a(r, c, zip(~T.f32_add, r, c, a, b))