torch.Tensor.multinomial

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

请参阅 torch.multinomial()