tfp.experimental.substrates.jax.math.log_combinations

View source on GitHub

Multinomial coefficient.

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


where i runs over all k classes.

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).

log_combinations Tensor representing the multinomial coefficient between n and counts.