jax.scipy.special.digamma

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

The digamma function.

LAX-backend implementation of psi(). Original docstring below.

psi(x, /, out=None, *, where=True, casting=’same_kind’, order=’K’, dtype=None, subok=True[, signature, extobj])

psi(z, out=None)

The logarithmic derivative of the gamma function evaluated at z.

Parameters
  • z (array_like) – Real or complex argument.

  • out (ndarray, optional) – Array for the computed values of psi.

Returns

digamma – Computed values of psi.

Return type

ndarray

Notes

For large values not close to the negative real axis psi is computed using the asymptotic series (5.11.2) from 1. For small arguments not close to the negative real axis the recurrence relation (5.5.2) from 1 is used until the argument is large enough to use the asymptotic series. For values close to the negative real axis the reflection formula (5.5.4) from 1 is used first. Note that psi has a family of zeros on the negative real axis which occur between the poles at nonpositive integers. Around the zeros the reflection formula suffers from cancellation and the implementation loses precision. The sole positive zero and the first negative zero, however, are handled separately by precomputing series expansions using 2, so the function should maintain full accuracy around the origin.

References

1(1,2,3)

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/