ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

Module: tf_agents.utils.numpy_storage

NumpyStorage stores nested objects across multiple numpy arrays.


class NumpyState: A checkpointable object whose NumPy array attributes are saved/restored.

class NumpyStorage: A class to store nested objects in a collection of numpy arrays.