tf_agents.utils.nest_utils.stack_nested_tensors

Stacks a list of nested tensors along the dimension specified.

tensors A list of nested tensors to be stacked.
axis the axis along which the stack operation is applied.

A stacked nested tensor.