Function squeeze 

pub fn squeeze<I, R, T, B, D>(
    tensor: &TensorBase<Storage<R, T, B>, D>,
    axes: I,
) -> TensorBase<Storage<DataRef<'_, <B as DeviceRawAPI<T>>::Raw>, T, B>, Vec<usize>>
where
    D: DimAPI,
    I: TryInto<AxesIndex<isize>, Error = Error>,
    R: DataAPI<Data = <B as DeviceRawAPI<T>>::Raw>,
    B: DeviceAPI<T>,

Removes singleton dimensions (axes) from x.

See also

Python array API standard: squeeze