PyTorch alternative for tf.data.experimental.sample_from_datasets

Question

Suppose I have two datasets, dataset one with 100 items and dataset two with 5000 items.

Now I want that during training my model sees as much items from dataset one