tf_agents.utils.composite.squeeze

Stay organized with collections Save and categorize content based on your preferences.

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.