tf_agents.utils.nest_utils.unstack_nested_arrays

Stay organized with collections Save and categorize content based on your preferences.

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.