Current section

Files

Jump to
viva_tensor include viva_tensor@nn@optim_AdamState.hrl
Raw

include/viva_tensor@nn@optim_AdamState.hrl

-record(adam_state, {
m :: viva_tensor@tensor:tensor(),
v :: viva_tensor@tensor:tensor(),
step :: integer()
}).