jax.scipy.stats.beta.pdf#

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

Beta probability distribution function.

JAX implementation of scipy.stats.beta pdf.

The pdf of the beta function is:

\[f(x, a, b) = \frac{\Gamma(a + b)}{\Gamma(a)\Gamma(b)} x^{a-1}(1-x)^{b-1}\]

where \(\Gamma\) is the gamma() function. It is defined for \(0\le x\le 1\) and \(b>0\).

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:

Array

jax.scipy.stats.beta.pdf