tf_agents.utils.nest_utils.unstack_nested_arrays

Unstack/unbatch a nest of numpy arrays.

tf_agents.utils.nest_utils.unstack_nested_arrays(
    nested_array
)

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

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