tf_agents.utils.nest_utils.get_outer_array_shape

View source on GitHub

Batch dims of array's batch dimension dim.