Module: tf_agents.utils.numpy_storage

View source on GitHub

NumpyStorage stores nested objects across multiple numpy arrays.

Classes

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.