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

tf_agents.utils.nest_utils.stack_nested_tensors

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.