docs menu

Module Vega.Adagrad_state

The structure of Adagrad states. Its walk visits sum_of_squares at that field as a position of the parameter, then step at "step" as a fixed int32 tensor. Transformations take adagrad_ptree.

#type ('p, 'd) t = ('p, 'd) adagrad_state

The type for values with parameter 'a and fixed arrays on 'd.

#val walk : ('a, 'b, 'd, 'e) Nx.Ptree.Walk.cursor -> ('a, 'd) t -> ('b, 'e) t

walk c x walks every part of x with c and is x rebuilt from what the walks return.