Fix output dimensions bug in hamiltonian_expand

[astralcai]

Context:
hamiltonian_expand reverses the order of dimensions for shot vectors and parameter broadcasting

Description of the Change:
Change a transpose operation to only transpose the last two dimensions if applicable.

Related GitHub Issues:
Fixes #5493
[sc-61037]

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 99.66%. Comparing base (a47d9bc) to head (314a2c9).
Report is 1 commits behind head on master.

Additional details and impacted files

@@            Coverage Diff             @@
##           master    #5494      +/-   ##
==========================================
- Coverage   99.67%   99.66%   -0.01%     
==========================================
  Files         406      406              
  Lines       37881    37622     -259     
==========================================
- Hits        37758    37497     -261     
- Misses        123      125       +2     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[lillian542]

My only real comment is that I'd like it if the tests checked the result, not only the result shape, since that doesn't really seem to be covered anywhere for the broadcasting branch of the code.

[lillian542]

Thanks, looks good to me @astralcai!