ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

tf_agents.utils.nest_utils.unstack_nested_tensors

Make list of unstacked nested tensors.

tensors Nested tensors whose first dimension is to be unstacked.
specs Tensor specs for tensors.

A list of the unstacked nested tensors.

ValueError if the tensors and specs have incompatible dimensions or shapes.