def sgd_l

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

import bend-ml-tensor-array@0.1.19.0/main.bend as BendMlTensorArray

import Base

342def sgd_l(nleft: Nat, +t: U32, +lr: F32, rw: Array<F32> & F32, rd: Array<F32> & F32) -> P2:
343  match nleft rw rd:
344    case 0n Tuple{w, x} Tuple{d, y}:
345      P2{w, d}
346    case 1n+p Tuple{w, +x} Tuple{d, +y}:
347      sgd_l(p, (t + 1 : U32), lr, Array.get(F32, Array.set(F32, w, t, (x - (lr * y : F32) : F32)), (t + 1 : U32)), Array.get(F32, d, (t + 1 : U32)))

Uses (12)

Used by (2)