guilhem_lavaux/JaxPM
0
0
Fork 1
mirror of https://github.com/DifferentiableUniverseInitiative/JaxPM.git synced 2025-06-29 16:41:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

update compensate CIC

Browse source
This commit is contained in:
Wassim KABALAN 2024-10-22 17:30:20 -04:00
parent cf799b6520
commit 0bb992fc56
2 changed files with 5 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

2
jaxpm/kernels.py
View file

@ -126,7 +126,7 @@ def cic_compensation(kvec):
wts: array
Complex kernel values
"""
kwts = [np.sinc(kvec[i] / (2 * np.pi)) for i in range(3)]
kwts = [jnp.sinc(kvec[i] / (2 * np.pi)) for i in range(3)]
wts = (kwts[0] * kwts[1] * kwts[2])**(-2)
return wts