jax.scipy.special.digamma

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

The digamma function.

LAX-backend implementation of psi().

Original docstring below.

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

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/