jax.scipy.stats.poisson.pmf#

jax.scipy.stats.poisson.pmf(k, mu, loc=0)[source]#

Probability mass function at k of the given RV.

LAX-backend implementation of scipy.stats._distn_infrastructure.pmf().

Original docstring below.

Parameters:

  • k (array_like) – Quantiles.

  • arg1 (array_like) – The shape parameter(s) for the distribution (see docstring of the instance object for more information)

  • arg2 (array_like) – The shape parameter(s) for the distribution (see docstring of the instance object for more information)

  • arg3 (array_like) – The shape parameter(s) for the distribution (see docstring of the instance object for more information)

  • ... (array_like) – The shape parameter(s) for the distribution (see docstring of the instance object for more information)

  • loc (array_like, optional) – Location parameter (default=0).

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

Returns:

pmf – Probability mass function evaluated at k

Return type:

array_like