tf_agents.specs.array_spec.add_outer_dims_nest