def sgd

bend-ml-autograd@0.1.1.0 · main.bend:295

import Base
import bend-ml-tensor@0.1.0.0/main.bend as T

295def sgd(-r: Nat, -c: Nat, +lr: F32, w: T.Mat<r, c>, dw: T.Mat<r, c>) -> T.Mat<r, c>:
296  T.Mat.sub(r, c, w, T.Mat.scale(r, c, lr, dw))