jax.scipy.stats.multinomial.pmf#
- jax.scipy.stats.multinomial.pmf(x, n, p)[source]#
Multinomial probability mass function.
LAX-backend implementation of
scipy.stats._multivariate.pmf()
.Original docstring below.
- Parameters:
x (array_like) – Quantiles, with the last axis of x denoting the components.
n (int) – Number of trials
p (array_like) – Probability of a trial falling into each category; should sum to 1
- Returns:
pmf – Probability density function evaluated at x
- Return type:
ndarray or scalar