jax.scipy.stats.multinomial.logpmf#

jax.scipy.stats.multinomial.logpmf(x, n, p)[源代码]#

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

scipy.stats.multinomial logpdf 的 JAX 实现。

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

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

其中 \(n = \sum_i x_i\)

参数:
  • x (

  • n (

  • p (

返回值:

对数 PMF 值的数组。

返回类型:

Array