class

ML::Optim::AdamState

Inherits Reference / Object

Adam optimizer state for a single parameter

Constructors

new(shape : Shape, device : Tensor::Device)

Instance methods

m

m=(m : Tensor)

step

step=(step : Int32)

v

v=(v : Tensor)