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