Have a question? Connect with the community at the TensorFlow Forum Visit Forum

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.