jax.scipy.special.digamma

jax.scipy.special.digamma(x)

The digamma function

JAX implementation of scipy.special.digamma.

\[\mathrm{digamma}(z) = \psi(z) = \frac{\mathrm{d}}{\mathrm{d}z}\log \Gamma(z)\]

where \(\Gamma(z)\) is the gamma() function.

Parameters:

x (jax.typing.ArrayLike) – arraylike, real-valued.

Returns:

array containing values of the digamma function.

Return type:

Array

Notes

The JAX version of digamma accepts real-valued inputs.

See also

• jax.scipy.special.gamma()

• jax.scipy.special.polygamma()