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

tf_agents.utils.nest_utils.get_outer_shape

Runtime batch dims of tensor's batch dimension dim.

nested_tensor Nest of tensors.
spec The nested spec.

A Tensor containing the outer shape.

ValueError If nested_tensor and spec have different structures.
TypeError If nested_tensor and spec structures have differing types.