jax.scipy.stats.truncnorm.logpdf#

jax.scipy.stats.truncnorm.logpdf(x, a, b, loc=0, scale=1)[source]#

截断正态对数概率分布函数。

scipy.stats.truncnorm logpdf 的 JAX 实现。

截断正态概率分布由下式给出

\[\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 值的数组。