jax.scipy.stats.dirichlet.pdf#
- jax.scipy.stats.dirichlet.pdf(x, alpha)[source]#
The Dirichlet probability density function.
LAX-backend implementation of
scipy.stats._multivariate.pdf().Original docstring below.
- Parameters:
x (array_like) – Quantiles, with the last axis of x denoting the components.
alpha (array_like) – The concentration parameters. The number of entries determines the dimensionality of the distribution.
- Returns:
pdf – The probability density function evaluated at x.
- Return type:
ndarray or scalar