ukiaddlZddlmZddlmZddlmZddlmZddlm Z ddl m Z m Z ddl mZed Zed ed e fd Zed ed e fd Zed ed e fdZed ed e fdZed ede d e fdZy)N)core)dtypes)lax) lax_numpy)ufuncs)Array ArrayLike) set_modulez jax.numpyMreturnctjt|d}tj}|dkrt j |fd|St j||}ddtjdtjz|z|dz z zz dtjdtjz|z|dz z zzS)a_Return a Blackman window of size M. JAX implementation of :func:`numpy.blackman`. Args: M: The window size. Returns: An array of size M containing the Blackman window. Examples: >>> with jnp.printoptions(precision=2, suppress=True): ... print(jnp.blackman(4)) [-0. 0.63 0.63 -0. ] See also: - :func:`jax.numpy.bartlett`: return a Bartlett window of size M. - :func:`jax.numpy.hamming`: return a Hamming window of size M. - :func:`jax.numpy.hanning`: return a Hanning window of size M. - :func:`jax.numpy.kaiser`: return a Kaiser window of size M. zM argument of jnp.blackmangzG??g{Gz? rconcrete_or_errorintrdefault_float_dtyperfulliotarcosnppir dtypens Z/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jax/_src/numpy/window_functions.pyblackmanrs. S!%AB!  $ $ &%!V 88QD!U ## hhua! fjjRUUQ!a%!899 9D6::aRTRWRWiZ[m_`cd_dNeCf>> with jnp.printoptions(precision=2, suppress=True): ... print(jnp.bartlett(4)) [0. 0.67 0.67 0. ] See also: - :func:`jax.numpy.blackman`: return a Blackman window of size M. - :func:`jax.numpy.hamming`: return a Hamming window of size M. - :func:`jax.numpy.hanning`: return a Hanning window of size M. - :func:`jax.numpy.kaiser`: return a Kaiser window of size M. zM argument of jnp.bartlettrr) rrrrrrrrrabsrs rbartlettr#;s|. S!%AB!  $ $ &%!V 88QD!U ## hhua! VZZA A &!a% 0 00r c0tjt|d}tj}|dkrt j |fd|St j||}ddtjdtjz|z|dz z zz S)aYReturn a Hamming window of size M. JAX implementation of :func:`numpy.hamming`. Args: M: The window size. Returns: An array of size M containing the Hamming window. Examples: >>> with jnp.printoptions(precision=2, suppress=True): ... print(jnp.hamming(4)) [0.08 0.77 0.77 0.08] See also: - :func:`jax.numpy.bartlett`: return a Bartlett window of size M. - :func:`jax.numpy.blackman`: return a Blackman window of size M. - :func:`jax.numpy.hanning`: return a Hanning window of size M. - :func:`jax.numpy.kaiser`: return a Kaiser window of size M. zM argument of jnp.hammingrgHzG?gq= ףp?rrrs rhammingr%Zs. S!%@A!  $ $ &%!V 88QD!U ## hhua! vzz!bee)a-1q5"9:: ::r c0tjt|d}tj}|dkrt j |fd|St j||}ddtjdtjz|z|dz z z zS)aYReturn a Hanning window of size M. JAX implementation of :func:`numpy.hanning`. Args: M: The window size. Returns: An array of size M containing the Hanning window. Examples: >>> with jnp.printoptions(precision=2, suppress=True): ... print(jnp.hanning(4)) [0. 0.75 0.75 0. ] See also: - :func:`jax.numpy.bartlett`: return a Bartlett window of size M. - :func:`jax.numpy.blackman`: return a Blackman window of size M. - :func:`jax.numpy.hamming`: return a Hamming window of size M. - :func:`jax.numpy.kaiser`: return a Kaiser window of size M. zM argument of jnp.hanningrrrrrs rhanningr'ys. S!%@A!  $ $ &%!V 88QD!U ## hhua! FJJq255y1}A677 88r betacptjt|d}tj}|dkrt j |fd|St j||}d|dz z}tj|tjd||z |z dzz ztj|z S)aReturn a Kaiser window of size M. JAX implementation of :func:`numpy.kaiser`. Args: M: The window size. beta: The Kaiser window parameter. Returns: An array of size M containing the Kaiser window. Examples: >>> with jnp.printoptions(precision=2, suppress=True): ... print(jnp.kaiser(4, 1.5)) [0.61 0.95 0.95 0.61] See also: - :func:`jax.numpy.bartlett`: return a Bartlett window of size M. - :func:`jax.numpy.blackman`: return a Blackman window of size M. - :func:`jax.numpy.hamming`: return a Hamming window of size M. - :func:`jax.numpy.hanning`: return a Hanning window of size M. zM argument of jnp.kaiserrrr) rrrrrrrrri0rsqrt)r r(rralphas rkaiserr-s0 S!%?@!  $ $ &%!V 88QD!U ## hhua! Q-% dV[[q5yE.Aa-G)GHH IILLY]L^ ^^r )numpyrjax._srcrr jax._src.laxrjax._src.numpyrrjax._src.typingrr jax._src.utilr exportrrr#r%r'r-r rr6s$!,$ K gggg<1111<;s;u;;<9s9u99<_c__u__r