tf_agents.utils.nest_utils.unstack_nested_tensors

Stay organized with collections Save and categorize content based on your preferences.

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.