torch.multinomial

The spelled-out intro to language modeling: building makemore

Andrej Karpathy

PyTorch function used to sample indices from a multinomial distribution (draw next character according to probabilities).