Train test split in torch library

I would like to shuffle my data (without dataloader).

Is there a function like in from sklearn.model_selection import train_test_split?

I would probably just use train_test_split directly to split the data indices and create Subsets using these shuffled and split indices.