Hi @Brando_Miranda, did you solve your problem already? When I saw your last question what came into my mind was to extract the data from cifar10 as a numpy array, like this
data = cifar10.train_data

Checkout the source code here 296. Now you can work with this array, use your rule to get the indices you want and pass them do SubsetRandomSampler.