Function dims2 

pub fn dims2(a_shape: &(usize, usize), b_shape: &(usize, usize)) -> (i32, i32)

Make sure that matrix shapes are compatible with A * B, and return the dimensions (m, n) safely cast to i32