TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tfp.experimental.substrates.numpy.math.log_combinations

View source on GitHub

Multinomial coefficient.

Aliases:

tfp.experimental.substrates.numpy.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.