Skip to main content
Ctrl+K

jax.lax.polygamma#

jax.lax.polygamma(m, x)[source]#

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

previous

jax.lax.pad

next

jax.lax.pow