tf_agents.utils.nest_utils.get_outer_shape

Runtime batch dims of tensor's batch dimension dim.