That’s the incorrect answer you get if you just directly reshape to the required shape without a transpose. That doesn’t work because it scrambles the data as explained on this thread.
So your previous code was actually closer, other than that you had the dimensions reversed.