They explained in the instructions why it does not work to just reshape directly to the final shape you want. It does not preserve the “channels” dimension. Here’s a thread which explains this in more detail. Also check the earlier post on that thread.
BTW this is not the Transfer Learning assignment, but the Neural Style Transfer one, meaning that the title of the thread is misleading.