Google I/O is a wrap! Catch up on TensorFlow sessions View sessions

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.