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

1

NIST Digital Library of Mathematical Functions https://dlmf.nist.gov/5

2

Fredrik Johansson and others. “mpmath: a Python library for arbitrary-precision floating-point arithmetic” (Version 0.19) http://mpmath.org/

Parameters

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