jax.scipy.stats.dirichlet.logpdf

内容

jax.scipy.stats.dirichlet.logpdf#

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

狄利克雷对数概率密度函数。

JAX 实现 scipy.stats.dirichlet logpdf

狄利克雷概率密度函数为

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

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

参数:
返回:

logpdf 值的数组。

返回类型:

数组