如何从PyTorch的FashionMNIST数据集中只获取特定的类？

import torch
from torchvision import datasets, transforms

transform = transforms.Compose([
    transforms.ToTensor(),
    transforms.Normalize((0.5,), (0.5,))
])

trainset = datasets.FashionMNIST('path_to_data', download=True, train=True, transform=transform)
testset = datasets.FashionMNIST('path_to_data', download=True, train=False, transform=transform)

def get_specific_class_data(dataset, class_label):
    indices = torch.where(dataset.targets == class_label)[0]
    return torch.utils.data.Subset(dataset, indices)

specific_class_data = get_specific_class_data(trainset, class_label)

batch_size = 64
specific_class_dataloader = torch.utils.data.DataLoader(specific_class_data, batch_size=batch_size, shuffle=True)