W2_A2_Ex-2_Flattening of the images

Please take a more careful look at the example code that they give you. What you have done is the equivalent of using X[0] where they are telling you to use X.shape[0]. See the difference?