jax.scipy.special.jn

Contents

jax.scipy.special.jn#

jax.scipy.special.jn(n, z)[source]#

Bessel function of the first kind of integer order n.

JAX implementation of scipy.special.jn.

\[J_n(z) = \sum_{k=0}^{\infty} \frac{(-1)^k}{k!\,(n + k)!} \left(\frac{z}{2}\right)^{2k + n}\]

for integer order \(n\) (with \(J_{-n}(z) = (-1)^n J_n(z)\)).

Parameters:

  • n (int) – The integer order of the Bessel function (must be a static integer).

  • z (ArrayLike) – The argument(s) at which to evaluate \(J_n(z)\). Real and complex inputs are supported.

Returns:

An array containing the evaluated Bessel function values.

Return type:

Array

Notes

Unlike scipy.special.jn, jax.scipy.special.jn requires n to be a concrete integer order (so n cannot be traced or batched with jax.vmap()).

See also

Contents