def lnb_b2

bendtorch@1.3.0.0 · main.bend:2065

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

2065def lnb_b2(+n: Nat, +i: Nat, +o: Nat, +par: Nat, x: TA.Mat<n, i>, g: TA.Mat<o, i>, r: TA.MMul<n, o, i>) -> Bwd<TA.Mat<o, i>, TA.Mat<n, i>>:
2066  match r:
2067    case TA.MMul{dy, w, dx}:
2068      lnb_b3(n, i, o, w, dx, TA.Mat.matmul_tn_acc(o, n, i, par, dy, x, g))

Uses (6)

Used by (1)

bendtorch@1.3.0.0/main:LinearNB