jax.scipy.stats.poisson.logpmf

jax.scipy.stats.poisson.logpmf(k, mu, loc=0)

Poisson log probability mass function.

JAX implementation of scipy.stats.poisson logpmf.

The Poisson probability mass function is given by

\[f(k) = e^{-\mu}\frac{\mu^k}{k!}\]

and is defined for \(k \ge 0\) and \(\mu \ge 0\).

Parameters:

• k (Array | ndarray | bool_ | number | bool | int | float | complex) – arraylike, value at which to evaluate the PMF

• mu (Array | ndarray | bool_ | number | bool | int | float | complex) – arraylike, distribution shape parameter

• loc (Array | ndarray | bool_ | number | bool | int | float | complex) – arraylike, distribution offset parameter

Returns:

array of logpmf values.

Return type:

Array

See also

• jax.scipy.stats.poisson.cdf()

• jax.scipy.stats.poisson.pmf()