def acti_f3

bendtorch-fast@1.3.0.0 · main.bend:285

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

285def acti_f3(~f: @+a: F32 -> F32, +n: Nat, +c: Nat, m: T.Two<TA.Mat<n, c>, TA.Mat<n, c>>) -> T.Fwd<Unit, TA.Mat<n, c>, TA.Mat<n, c>>:
286  match m:
287    case T.Two{x, s}:
288      T.Fwd{Unit{}, map(~f, n, c, x), s}