jax.scipy.stats.multinomial.logpmf

内容

jax.scipy.stats.multinomial.logpmf#

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

多项式对数概率质量函数。

JAX 实现 scipy.stats.multinomial logpdf

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

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

其中 \(n = \sum_i x_i\)

参数:
返回:

logpmf 值的数组。

返回类型:

数组