Save the date! Google I/O returns May 18-20 Register now

tf_agents.utils.nest_utils.get_outer_array_shape

Batch dims of array's batch dimension dim.