ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more


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.
num_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 num_outer_dims.