def Mat.sgd

bend-ml-tensor-array@0.1.19.0 · main.bend:1357

import Base

1357def Mat.sgd(+r: Nat, +c: Nat, +lr: F32, w: Mat<r, c>, dw: Mat<r, c>) -> MPair<r, c>:
1358  match w dw:
1359    case Mat{x} Mat{y}:
1360      mpair_fin(r, c, sgd_l(Nat.mul(r, c), 0, lr, Array.get(F32, x, 0), Array.get(F32, y, 0)))