jax.numpy.i0

jax.numpy.i0(x)

Calculate modified Bessel function of first kind, zeroth order.

JAX implementation of numpy.i0().

Modified Bessel function of first kind, zeroth order is defined by:

\[\mathrm{i0}(x) = I_0(x) = \sum_{k=0}^{\infty} \frac{(x^2/4)^k}{(k!)^2}\]

Parameters:

x (ArrayLike) – scalar or array. Specifies the argument of Bessel function. Complex inputs are not supported.

Returns:

An array containing the corresponding values of the modified Bessel function of x.

Return type:

Array

See also

• jax.scipy.special.i0(): Calculates the modified Bessel function of zeroth order.

• jax.scipy.special.i1(): Calculates the modified Bessel function of first order.

• jax.scipy.special.i0e(): Calculates the exponentially scaled modified Bessel function of zeroth order.

Examples

>>> x = jnp.array([-2, -1, 0, 1, 2])
>>> jnp.i0(x)
Array([2.2795851, 1.266066 , 1.0000001, 1.266066 , 2.2795851], dtype=float32)