tf_agents.utils.nest_utils.unstack_nested_arrays

View source on GitHub

Unstack/unbatch a nest of numpy arrays.

nested_array Nest of numpy arrays where each array has shape [batch_size, ...].

A list of length batch_size where each item in the list is a nest having the same structure as nested_array.