jax.scipy.stats.multinomial.pmf

内容

jax.scipy.stats.multinomial.pmf#

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

多项式概率质量函数。

JAX 实现 scipy.stats.multinomial pmf

多项式概率分布由下式给出:

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

其中 \(n = \sum_i x_i\).

参数:
返回值:

pmf 值的数组

返回类型:

数组