jax.scipy.stats.poisson.pmf#
- jax.scipy.stats.poisson.pmf(k, mu, loc=0)[source]#
Poisson probability mass function.
JAX implementation of
scipy.stats.poisson
pmf
.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 pmf values.
- Return type: