If Saif’s suggestions are not enough to get you there, the best way to start diagnosing a case like this is to understand the “dimensional analysis”, which will show you what the shapes are that you should be seeing at each layer. Here’s a thread which walks you through that for this test case.