torch.Tensor.multinomial

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

参见 torch.multinomial()