samples=torch.multinomial(probs,5)
but samples data may be duplicate
TLDR: use WeightedRandomSampler with replace=False
WeightedRandomSampler
replace=False