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