from torch.utils.data import random_split code example

Example: torch.utils.data.random_split(dataset, lengths)

>>> random_split(range(10), [3, 7], generator=torch.Generator().manual_seed(42))