Python Tensor To Numpy Array Dimensions