Fix/shape dimension order

[MaxFBurg]

Addresses #175

[codecov]

Codecov Report

Attention: Patch coverage is 0% with 56 lines in your changes are missing coverage. Please review.

Project coverage is 7.53%. Comparing base (9b9ab51) to head (28aedec).

Files	Patch %	Lines
neuralpredictors/layers/readouts/gaussian.py	0.00%	56 Missing ⚠️

Additional details and impacted files

@@          Coverage Diff          @@
##            main    #225   +/-   ##
=====================================
  Coverage   7.52%   7.53%           
=====================================
  Files         58      58           
  Lines       5965    5959    -6     
  Branches    1011    1011           
=====================================
  Hits         449     449           
+ Misses      5479    5473    -6     
  Partials      37      37           

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

[pollytur]

Sequence might cause problems for python below 3.6 but I believe that's neglectable in 2024:)