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