~/jendhub

def lin_b2

bendtorch-fast@1.3.0.0 · main.bend:250

import bendtorch-fast@1.3.0.0/main.bend as BendtorchFast
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

250def lin_b2(+n: Nat, +i: Nat, +o: Nat, par: Nat, b: TA.Mat<1n, o>, x: TA.Mat<n, i>, g: T.Lin<i, o>, r: TA.MMul<n, o, i>) -> T.Bwd<T.Lin<i, o>, TA.Mat<n, i>>:
251  match g r:
252    case T.Lin{gw, gb} TA.MMul{dy, w, dx}:
253      T.lin_b3(n, i, o, w, b, dx, gb, TF.matmul_tn_acc(o, n, i, par, dy, x, gw))

Uses (7)

Used by (1)