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

tf_agents.utils.nest_utils.stack_nested_arrays

Stack/batch a list of nested numpy arrays.

nested_arrays A list of nested numpy arrays of the same shape/structure.

A nested array containing batched items, where each batched item is obtained by stacking corresponding items from the list of nested_arrays.