jax.numpy.i0

jax.numpy.i0(x)

計算第一類零階修正貝索函數。

JAX 實作的 numpy.i0()。

第一類零階修正貝索函數定義為

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

參數:

x (ArrayLike) – 純量或陣列。指定貝索函數的引數。不支援複數輸入。

傳回:

包含 x 對應修正貝索函數值的陣列。

傳回類型:

陣列 (Array)

另請參閱

• jax.scipy.special.i0()：計算零階修正貝索函數。

• jax.scipy.special.i1()：計算第一階修正貝索函數。

• jax.scipy.special.i0e()：計算指數縮放的零階修正貝索函數。

範例

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