tf.contrib.distributions.Dirichlet.prob()

tf.contrib.distributions.Dirichlet.prob(value, name='prob')

Probability density/mass function (depending on is_continuous).

Additional documentation from Dirichlet:

Note that the input must be a non-negative tensor with dtype dtype and whose shape can be broadcast with self.alpha. For fixed leading dimensions, the last dimension represents counts for the corresponding Dirichlet distribution in self.alpha. x is only legal if it sums up to one.

Args:

• value: float or double Tensor.
• name: The name to give this op.

Returns:

• prob: a Tensor of shape sample_shape(x) + self.batch_shape with values of type self.dtype.