jax.lax.polygamma

jax.lax.polygamma(m, x)

Elementwise polygamma: \(\psi^{(m)}(x)\).

Parameters:

• m (Union[Array, ndarray, bool_, number, bool, int, float, complex])

• x (Union[Array, ndarray, bool_, number, bool, int, float, complex])

Return type:

Array