Trying to understand the torch.multinomial

6

Yes, I think your understanding is correct.
If I’m not mistaken, the weights are simply normalized such that their sum equals to 1.