Just reshaping directly to the final shape may or may not give you what you really want. There was an example of this in Course 1 Week 2 when we unrolled the images. Here’s a thread which explains the details in that case. The same idea applies here: if you just directly reshape, you end up scrambling the data. It is essential that you preserve the “channels” dimension in the output.
3 Likes