Function is_square 

pub fn is_square<const D: usize>(shape: &[(Dimension, usize); D]) -> bool

Returns true if the dimensions are all the same length. For 0 or 1 dimensions trivially returns true. For 2 dimensions, this corresponds to a square matrix, and for 3 dimensions, a cube shaped tensor, and so on.