jax.scipy.stats.truncnorm.logpdf#
- jax.scipy.stats.truncnorm.logpdf(x, a, b, loc=0, scale=1)[source]#
截断正态分布的对数概率密度函数。
JAX 实现
scipy.stats.truncnorm的logpdf。截断正态概率分布定义为:
\[\begin{split}f(x, a, b) = \begin{cases} \frac{1}{\sqrt{2\pi}}e^{-x^2/2} & a \le x \le b \\ 0 & \mathrm{otherwise} \end{cases}\end{split}\]其中 \(a\) 和 \(b\) 是从零开始计算的标准差。JAX 使用 scipy 的术语,将
loc表示均值,scale表示标准差。- 参数:
x – 数组型,用于评估 PDF 的值
a – 类似数组,分布形状参数
b – 类数组,分布形状参数
loc – 类数组,分布偏移参数
scale – 类数组,分布比例参数
- 返回:
logpdf 值的数组。