Ah, I see, it does.... you reshape to get back to a square grid, then transpose to change the order for convolution in the next step.  That's fine.