jax.scipy.stats.multinomial.pmf

Contents

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