GitHub

tf_agents.utils.nest_utils.stack_nested_arrays

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.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2021-05-21 UTC.

Args