jax.scipy.stats.dirichlet.logpdf¶

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

Log of the Dirichlet probability density function.

LAX-backend implementation of logpdf().

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 – Log of the probability density function evaluated at x.

Return type

ndarray or scalar