jax.scipy.special.polygamma

jax.scipy.special.polygamma(n, x)[source]

Polygamma functions.

LAX-backend implementation of polygamma().

Original docstring below.

Defined as \(\psi^{(n)}(x)\) where \(\psi\) is the digamma function. See [dlmf] for details.

Parameters
  • n (array_like) – The order of the derivative of the digamma function; must be integral

  • x (array_like) – Real valued input

Returns

Function results

Return type

ndarray

References

dlmf

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