jax.scipy.stats.multinomial.logpmf

jax.scipy.stats.multinomial.logpmf(x, n, p)

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

JAX implementation of 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\)。

参数:

• x (Array | ndarray | bool | number | bool | int | float | complex) – arraylike, 用于评估PMF的值

• n (Array | ndarray | bool | number | bool | int | float | complex) – arraylike, 分布形状参数

• p (Array | ndarray | bool | number | bool | int | float | complex) – arraylike, 分布形状参数

返回:

logpmf 值的数组。

返回类型:

Array

参见

jax.scipy.stats.multinomial.pmf()