Mnist create 1-d vector from image matrix

how do I convert the image matrix of minst dataset to 1-d Vector ? after loading the minst dataset and applying transform=toTensor() ??

train_dataset = torchvision.datasets.MNIST(root='./data',train=True, transform=transforms.ToTensor(), download=True)
test_dataset = torchvision.datasets.MNIST(root='./data',train=False, transform=transforms.ToTensor(),download = True)

train_loader = torch.utils.data.DataLoader(dataset=train_dataset, batch_size=4, shuffle=True)
test_loader = torch.utils.data.DataLoader(dataset=test_dataset, batch_size=4, shuffle=False)

you can use .view() to format the data as 1D vector.

iter = train_loader.__iter__()
images = iter.next() #4x1x28x28
images = images.view(images.shape[0], -1) # 4x784

3 Likes