jax.scipy.stats.dirichlet.logpdf

jax.scipy.stats.dirichlet.logpdf(x, alpha)

Dirichlet 对数概率分布函数。

JAX implementation of 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() 函数。

参数:

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

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

返回:

对数概率密度值的数组。

返回类型:

Array

参见

jax.scipy.stats.dirichlet.pdf()