ошибка: RuntimeError: неверный аргумент 0: размеры тензоров должны совпадать, кроме как в измерении 0. Получил 168 и 73 в измерении 2

Question

path = ("numberplates\plates\\")
dirs = os.listdir( path )

def resize():
    for item in dirs:
        if os.path.isfile(path+item):
            im = Image.open(path+item)
            f, e = os.path.splitext(path+item)
            imResize = im.resize((100,100), Image.ANTIALIAS)
            imResize.save(f + ' resized.jpg', 'JPEG', quality=90)

resize()
data_path = "numberplates"
train_dataset = torchvision.datasets.ImageFolder(root=data_path,transform=torchvision.transforms.ToTensor())
train_loader = torch.utils.data.DataLoader(train_dataset,batch_size=1636,num_workers=0,shuffle=True)
dataiter = iter(train_loader)
images, labels = dataiter.next()

print(images.shape)

print(images[0].shape)
print(labels[0].item())