Python Get Dimension Of Numpy Array Multiplication