jax.scipy.stats.truncnorm.logpdf#

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

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

JAX 实现 scipy.stats.truncnormlogpdf

截断正态概率分布定义为:

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