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.