jax.scipy.stats.dirichlet.logpdf#

jax.scipy.stats.dirichlet.logpdf(x, alpha)[源代码]#

狄利克雷对数概率分布函数。

scipy.stats.dirichlet logpdf 的 JAX 实现。

狄利克雷概率密度函数为

\[f(\mathbf{x}) = \frac{1}{B(\mathbf{\alpha})} \prod_{i=1}^K x_i^{\alpha_i - 1}\]

其中 \(B(\mathbf{\alpha})\)beta() 函数,它位于一个 \(K\) 维向量空间中。

参数:
返回:

logpdf 值的数组。

返回类型:

Array