RSVP for your your local TensorFlow Everywhere event today!

oryx.core.Pytree

Class that registers objects as Jax pytree_nodes.

Methods

flatten

View source

unflatten

View source