Google I/O returns May 18-20! Reserve space and build your schedule Register now

tf_agents.utils.nest_utils.get_outer_shape

Runtime batch dims of tensor's batch dimension dim.