ninetoothed.Tensor.squeeze

Tensor.squeeze(dim)

Removes the specified singleton dimensions of the tensor.

Parameters:

dim – The dimension(s) to be squeezed.

Returns:

The squeezed tensor.