【Bug】PyTorch DataLoader - “IndexError: too many indices for tensor of dimension 0“
importtorchimporttorchvisionimporttorchvision.transformsastransformstransform=transforms.Compose([transforms.ToTensor(),transforms.Normalize((0.5),(0.5))])trainset=torchvision.datasets.MNIST