strusov/JaxPM_highres
0
0
Fork 0
forked from Aquila-Consortium/JaxPM_highres
Code Pull requests Activity

fix power spec

Browse source
This commit is contained in:
EiffL 2022-03-27 04:39:07 +02:00
parent 10b093f07f
commit 3e1aa2d4e2
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
jaxpm/utils.py
View file

@ -67,8 +67,8 @@ def power_spectrum(field, kmin=5, dk=0.5, boxsize=False):
real = jnp.real(pk).reshape([-1])
imag = jnp.imag(pk).reshape([-1])
Psum = jnp.bincount(dig, weights=(W.flatten() * imag), minlength=xsum.size) * 1j
Psum += jnp.bincount(dig, weights=(W.flatten() * real), minlength=xsum.size)
Psum = jnp.bincount(dig, weights=(W.flatten() * imag), length=xsum.size) * 1j
Psum += jnp.bincount(dig, weights=(W.flatten() * real), length=xsum.size)
P = ((Psum / Nsum)[1:-1] * boxsize.prod()).astype('float32')