squeeze

axonml_tensor::shape

Function squeeze 

Source 
pub fn squeeze(shape: &[usize], dim: Option<usize>) -> Shape
Expand description

Computes the shape after squeezing (removing dimensions of size 1).

§Arguments

  • shape - Input shape
  • dim - Optional dimension to squeeze (None = all)

§Returns

Squeezed shape.