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

tf_agents.utils.nest_utils.unstack_nested_arrays

View source on GitHub

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.