tf_agents.utils.nest_utils.get_outer_array_shape

Batch dims of array's batch dimension dim.