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

tf_agents.utils.composite.squeeze

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.