astro-informatics / s2wav

Differentiable and accelerated wavelet transform on the sphere with JAX
https://astro-informatics.github.io/s2wav/
MIT License
14 stars 0 forks source link

Dimension Helper Functions #24

Closed JessWhitney closed 1 year ago

JessWhitney commented 1 year ago

Code for issue #8 . Includes three sections of code:

Some functions were also called from so3, in those cases rather than writing the entire general function I have only included the case needed.

I've also tried to include a brief description for each function, but for some of the functions towards the end I wasn't 100% sure on what they were doing, so I expect I will have to alter them after review.

JessWhitney commented 1 year ago

This incorrectly included the parameters file, so I'm going to make sure it's all branched correctly then resubmit a PR with only the dimension helper file.