multinomial¶
dragon.random.
multinomial
(
inputs,
sample_size=1,
**kwargs
)[source]¶Return an index tensor sampled from the multinomial distribution.
Examples:
input = dragon.math.log(dragon.constant([0.5, 0.5])) index = dragon.random.multinomial(input)
- Parameters:
- inputs (dragon.Tensor) – The input tensor.
- sample_size (int, optional, default=1) – The number of samples.
- Returns:
dragon.Tensor – The output tensor.