jax.scipy.special.digamma

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

The digamma function.

LAX-backend implementation of 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

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/