jax.scipy.stats.truncnorm.logpdf#

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

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

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{其他情况} \end{cases}\end{split}\]

其中\(a\)\(b\)实际上是以距零的标准差数量指定的。JAX使用scipy的命名法,用loc表示中心点,用scale表示标准差。

参数:
  • x – 类数组,用于评估PDF的值

  • a – 类数组,分布形状参数

  • b – 类数组,分布形状参数

  • loc – 类数组,分布偏移参数

  • scale – 类数组,分布尺度参数

返回值:

logpdf 值的数组。