tf_agents.utils.nest_utils.get_outer_shape

View source on GitHub

Runtime batch dims of tensor's batch dimension dim.