def lin_fwd

bendtorch@1.3.0.0 · main.bend:615

import bendtorch@1.3.0.0/main.bend as Bendtorch

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

615def lin_fwd(+n: Nat, +i: Nat, +o: Nat, par: Nat, p: Lin<i, o>, x: TA.Mat<n, i>) -> Fwd<Lin<i, o>, TA.Mat<n, o>, TA.Mat<n, i>>:
616  match p:
617    case Lin{w, b}:
618      lin_f2(n, i, o, b, TA.Mat.matmul_nt(n, i, o, par, x, w))

Uses (6)

Used by (2)