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. |