tfp.math.log_combinations

View source on GitHub

Multinomial coefficient.

tfp.math.log_combinations(
    n,
    counts,
    name='log_combinations'
)

Given n and counts, where counts has last dimension k, we compute the multinomial coefficient as:

n! / sum_i n_i!

where i runs over all k classes.

Args:

  • n: Floating-point Tensor broadcastable with counts. This represents n outcomes.
  • counts: Floating-point Tensor broadcastable with n. This represents counts in k classes, where k is the last dimension of the tensor.
  • name: A name for this operation (optional).

Returns:

  • log_combinations: Tensor representing the multinomial coefficient between n and counts.