jax.scipy.stats.multinomial.pmf#

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

多项分布的概率质量函数。

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值的数组

返回类型:

Array

另请参阅

jax.scipy.stats.multinomial.logpmf()