Hi, Fernando.
It’s great to hear that you were able to find the solution under your own power. Congrats!
For anyone else who sees this, in addition to the thread Fernando links above, there is also this thread which gives a specific example for the flattening that happens in the Art Generation with Neural Style Transfer assignment.