Function squeeze

Source
pub fn squeeze(x: &Tensor, dims: impl Dims) -> Tensor