torch.distributions.multinomial.Multinomial (an example mistake of docs(03) Save Image Image gallery: Torch Multinomial