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


Squeeze composite tensor along axis axis.

t A Tensor or SparseTensor.
axis A python integer.

The tensor with dimension axis removed.

InvalidArgumentError If t is a SparseTensor and has more than one index stored along axis.