jax.scipy.stats.dirichlet.pdf

内容

jax.scipy.stats.dirichlet.pdf#

jax.scipy.stats.dirichlet.pdf(x, alpha)[source]#

狄利克雷概率分布函数。

JAX 实现 scipy.stats.dirichlet pdf

狄利克雷概率密度函数为

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

其中 \(B(\mathbf{\alpha})\)\(K\) 维向量空间中的 beta() 函数。

参数:
返回值:

pdf 值数组。

返回类型:

Array