jax.scipy.stats.multinomial.logpmf

jax.scipy.stats.multinomial.logpmf#

jax.scipy.stats.multinomial.logpmf(x, n, p)[source]#

Multinomial log probability mass function.

JAX implementation of scipy.stats.multinomial logpdf.

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 logpmf values.

Return type:

Array