~/jendhub

def Train.step_at

bendtorch@1.3.0.0 · main.bend:968

import bendtorch@1.3.0.0/main.bend as Bendtorch
import Base

968def Train.step_at(-P: Type, -X: Type, -Y: Type, -S: Type, -Target: Type, m: Layer<P, X, Y, S>, loss: Y -> Target -> LossOut<Y>, +cfg: OptCfg, tr: Train<P>, x: X, t: Target) -> Train<P>:
969  match m tr:
970    case Layer{ff, bf, sf, zeros, dump, load, store, count} Train{p, g, st, p1, p2, l}:
971      train_at_f(P, X, Y, S, Target, bf, sf, loss, cfg, g, st, p1, p2, t, ff(p, x))

Uses (5)

Used by (0)

No other definition uses this definition.