tf_agents.specs.tensor_spec.remove_outer_dims_nest

Removes the specified number of outer dimensions from the input spec nest.

specs Nested list/tuple/dict of TensorSpecs/ArraySpecs, describing the shape of tensors.
outer_dims (int) Number of outer dimensions to remove.

Nested TensorSpecs with outer dimensions removed from the input specs.

Value error if a spec in the nest has shape rank less than outer_dims.