def lin_bwd

bendtorch-fast@1.3.0.0 · main.bend:255

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

255def lin_bwd(+n: Nat, +i: Nat, +o: Nat, +par: Nat, p: T.Lin<i, o>, x: TA.Mat<n, i>, dy: TA.Mat<n, o>, g: T.Lin<i, o>) -> T.Bwd<T.Lin<i, o>, TA.Mat<n, i>>:
256  match p:
257    case T.Lin{w, b}:
258      lin_b2(n, i, o, par, b, x, g, TF.matmul_precise(n, o, i, par, dy, w))