jax.scipy.stats.dirichlet.pdf¶

jax.scipy.stats.dirichlet.pdf(x, alpha)[source]¶

The Dirichlet probability density function.

LAX-backend implementation of 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