Solved by Oli in post #2 
              
              
                You are printing the wrong variable. print("After upconv, X1 shape:", x1.shape) should be print("After upconv, X shape:", x.shape)