jax.scipy.stats.multinomial.pmf#
- jax.scipy.stats.multinomial.pmf(x, n, p)[source]#
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:
See also