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\) 维向量空间中。