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

tf_agents.utils.nest_utils.unstack_nested_tensors

View source on GitHub

Make list of unstacked nested tensors.

tf_agents.utils.nest_utils.unstack_nested_tensors(
    tensors, specs
)

Args:

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

Returns:

A list of the unstacked nested tensors.

Raises:

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