As the indices are drawn using torch.multinominal I assume the only way to ensure all samples are selected is by setting replacement=False.
On the other hand you could try to set num_samples to such a number that your most unlikely class with be drawn with a specified probability.