I have a numpy array say x.shape (5,) x[0].shape (512,) How to apply reshape so that when I get x.shape (5,512)