View source on GitHub |
Stacks a list of nested tensors along the dimension specified.
tf_agents.utils.nest_utils.stack_nested_tensors(
tensors, axis=0
)
Args | |
---|---|
tensors
|
A list of nested tensors to be stacked. |
axis
|
the axis along which the stack operation is applied. |
Returns | |
---|---|
A stacked nested tensor. |