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

Module: tf_agents.networks.nest_map

Network layer that allows mapping multiple inputs.


class NestMap: The NestMap network processes nested inputs via nested layers.


NestFlatten(...): Returns a Keras layer that takes a nest of inputs, and returns a list.