jax.scipy.stats.truncnorm.pdf#
- jax.scipy.stats.truncnorm.pdf(x, a, b, loc=0, scale=1)[source]#
Truncated normal probability distribution function.
JAX implementation of
scipy.stats.truncnormpdf.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 the centroid. 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 pdf values.
- Return type: