jax.scipy.special.polygamma

Contents

jax.scipy.special.polygamma#

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

Polygamma functions.

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