jax.scipy.special.digamma

Contents

jax.scipy.special.digamma#

jax.scipy.special.digamma(x)[source]#

The digamma function.

LAX-backend implementation of scipy.special.psi().

The JAX version only accepts real-valued inputs.

Original docstring below.

The logarithmic derivative of the gamma function evaluated at z.

Returns:

digamma – Computed values of psi.

Return type:

scalar or ndarray

References

Parameters:

x (Union[Array, ndarray, bool_, number, bool, int, float, complex])