jax.scipy.stats.multinomial.pmf

jax.scipy.stats.multinomial.pmf(x, n, p)

Multinomial probability mass function.

JAX implementation of scipy.stats.multinomial pmf.

The multinomial probability distribution is given by

\[f(x, n, p) = n! \prod_{i=1}^k \frac{p_i^{x_i}}{x_i!}\]

with \(n = \sum_i x_i\).

Parameters:

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

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

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

Returns:

array of pmf values

Return type:

Array

See also

jax.scipy.stats.multinomial.logpmf()