pub fn dims3(
a_shape: &(usize, usize),
b_shape: &(usize, usize),
c_shape: &(usize, usize),
) -> (i32, i32, i32)Expand description
Make sure that matrix shapes are compatible with C = A * B, and
return the dimensions (m, n, k) safely cast to i32, where C is (m x n), and k is the common dimension of A and B