tf_agents.utils.nest_utils.unstack_nested_tensors

View source on GitHub

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.