MultinomialΒΆ

Generate a tensor of samples from a multinomial distribution according to the probabilities of each of the possible outcomes.

Abstract Signature:

Multinomial(input, dtype: int, sample_size: int, seed: float)

NumPy

API: numpy.random.mtrand.multinomial
Strategy: Direct Mapping

Flax NNX

API: nnx.rnglib.Rngs.multinomial
Strategy: Direct Mapping