Torch.utils.data.dataset.random_split code example
Example: torch.utils.data.random_split(dataset, lengths)
>>> random_split(range(10), [3, 7], generator=torch.Generator().manual_seed(42))
>>> random_split(range(10), [3, 7], generator=torch.Generator().manual_seed(42))