i am using
FashionMNIST_train_dataset = torchvision.datasets.FashionMNIST(
    root = './data/FashionMNIST',
    train = True,
    download = True,
    transform = transforms.Compose([
        transforms.ToTensor()                                 
    ])
)

and I want SPLIT this data to N balanced labeled data, and the rest (60000-N).
how can I do that?