Good to hear it’s working.
However, I’m a bit afraid when you say “resizing the input”, as a view or reshape won’t yield the desired output.
To swap the axes, use permute, if you haven’t already done so.