jax.scipy.stats.chi2.logpdf#
- jax.scipy.stats.chi2.logpdf(x, df, loc=0, scale=1)[源码]#
卡方(Chi-square)对数概率分布函数。
JAX 对
scipy.stats.chi2的logpdf的实现。卡方概率分布函数由以下公式给出:
\[\begin{split}f(x, k) = \begin{cases} \frac{x^{k/2-1}e^{-x/2}}{2^{k/2}\Gamma(k/2)} & x \ge 0 \\ 0 & \mathrm{otherwise} \end{cases}\end{split}\]其中 \(k\) 是自由度,\(\Gamma\) 是
gamma()函数。JAX 遵循 scipy 的约定,使用df来表示自由度。- 参数:
x (Array | ndarray | bool | number | bool | int | float | complex | TypedNdArray) – arraylike,计算 PDF 的值
df (Array | ndarray | bool | number | bool | int | float | complex | TypedNdArray) – arraylike,分布形状参数
loc (Array | ndarray | bool | number | bool | int | float | complex | TypedNdArray) – arraylike,分布偏移参数
scale (Array | ndarray | bool | number | bool | int | float | complex | TypedNdArray) – arraylike,分布尺度参数
- 返回:
logpdf 值的数组。
- 返回类型: