Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tf_agents.utils.nest_utils.stack_nested_arrays

View source on GitHub

Stack/batch a list of nested numpy arrays.

tf_agents.utils.nest_utils.stack_nested_arrays(
    nested_arrays
)

Args:

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

Returns:

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