guilhem_lavaux/JaxPM
0
0
Fork 1
mirror of https://github.com/DifferentiableUniverseInitiative/JaxPM.git synced 2025-05-14 03:51:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

creoss correlation function

Browse source
This commit is contained in:
denise lanzieri 2022-06-18 18:23:46 +02:00
parent 8b885450a8
commit 84b79af7f8
2 changed files with 49 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

1
jaxpm/pm.py
View file

@ -80,7 +80,6 @@ def pgd_correction(pos, mesh_shape, cosmo, params):
params: [alpha, kl, ks] pgd parameters
"""
kvec = fftk(mesh_shape)
delta = cic_paint(jnp.zeros(mesh_shape), pos)
alpha, kl, ks = params
delta_k = jnp.fft.rfftn(delta)