docs menu

Module Vega.Lion_state

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

#type ('p, 'd) t = ('p, 'd) lion_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.