jax.scipy.special.digamma#
- jax.scipy.special.digamma(x)[source]#
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 (ArrayLike) – arraylike, real-valued.
- Returns:
array containing values of the digamma function.
- Return type:
Notes
The JAX version of digamma accepts real-valued inputs.