add switch for partial analysis shift

astro-informatics / s2wav

Differentiable and accelerated wavelet transform on the sphere with JAX

https://astro-informatics.github.io/s2wav/

MIT License

12 stars 0 forks source link

add switch for partial analysis shift #81

Closed CosmoMatt closed 2 months ago

CosmoMatt commented 2 months ago

Extremely minor PR which just adds a switch to the shift introduced to precompute flm_to_analysis to avoid duplicating memory for highest wavelet scale, whilst being consistent for e.g. scattering.