jax.scipy.stats.truncnorm.logpdf

内容

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