only one element tensors can be converted to Python scalars while doing PCA for fashion mnist data after selecting 2 classes

Question

\'\'\'

# Selecting classes 0 and 8
idx = (trainset.targets==0) | (trainset.targets==8)
trainset.targets = trainset.targets[idx]
trainset.data = trainset.data[