jax.scipy.stats.truncnorm.logpdf#
- jax.scipy.stats.truncnorm.logpdf(x, a, b, loc=0, scale=1)[source]#
Truncated normal log probability distribution function.
JAX implementation of
scipy.stats.truncnormlogpdf.The truncated normal probability distribution is given by
\[\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}\]where \(a\) and \(b\) are effectively specified in number of standard deviations from zero. JAX uses the scipy nomenclature of
locfor the centroid andscalefor the standard deviation.- Parameters:
x (Array | ndarray | bool | number | bool | int | float | complex) – arraylike, value at which to evaluate the PDF
a (Array | ndarray | bool | number | bool | int | float | complex) – arraylike, distribution shape parameter
b (Array | ndarray | bool | number | bool | int | float | complex) – arraylike, distribution shape parameter
loc (Array | ndarray | bool | number | bool | int | float | complex) – arraylike, distribution offset parameter
scale (Array | ndarray | bool | number | bool | int | float | complex) – arraylike, distribution scale parameter
- Returns:
array of logpdf values.
- Return type: