~/jendhub

def Opt.sgd_fin

bendtorch@1.3.0.0 · main.bend:419

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

419# One matrix step, with only the buffers this optimizer reads.
420def Opt.sgd_fin(-r: Nat, -c: Nat, q: TA.P2) -> MatStep<r, c>:
421  match q:
422    case TA.P2{w, g}: MatStep{TA.Mat{w}, TA.Mat{g}, NoMom{}}

Uses (5)

Used by (1)