jax.scipy.special.polygamma#
- jax.scipy.special.polygamma(n, x)[source]#
The polygamma function.
JAX implementation of
scipy.special.polygamma().\[\mathrm{polygamma}(n, x) = \psi^{(n)}(x) = \frac{\mathrm{d}^n}{\mathrm{d}x^n}\log \Gamma(x)\]where \(\Gamma\) is the
gamma()function.- Parameters:
n (jax.typing.ArrayLike) – arraylike, integer-valued. The order of the derivative.
x (jax.typing.ArrayLike) – arraylike, real-valued. The value at which to evaluate the function.
- Returns:
array
- Return type: