torch.Tensor.multinomial

Tensor.multinomial(num_samples, replacement=False, *, generator=None) → Tensor

See torch.multinomial()