jax.scipy.special.iv

Contents

jax.scipy.special.iv#

jax.scipy.special.iv(v, z)[source]#

Modified Bessel function of the first kind of integer order v.

JAX implementation of scipy.special.iv for integer orders.

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

for integer order \(v \ge 0\) (with \(I_{-v}(z) = I_v(z)\)).

Parameters:

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

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

Returns:

An array containing the evaluated modified Bessel function values.

Return type:

Array

Notes

Unlike scipy.special.iv, jax.scipy.special.iv requires v to be a concrete integer order (so non-integer orders are not supported and v cannot be traced or batched with jax.vmap()).

See also

Contents