torch.Tensor.multinomial

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

請參閱 torch.multinomial()