momentumStep

momentumStep(x, gradient, state?, options?): object

Defined in: steps.js:49

Gradient descent with momentum: v = μ v + lr · g; x -= v.

Parameters

x

number[] | Float64Array<ArrayBufferLike>

moved in place

gradient

number[] | Float64Array<ArrayBufferLike>

state?

velocity

Float64Array

options?

StepOptions & object = {}

Returns

object

velocity

velocity: Float64Array