mirror of
https://github.com/luzhisheng/js_reverse.git
synced 2025-04-23 05:39:22 +08:00
11 lines
302 B
Python
11 lines
302 B
Python
from torchvision.datasets import MNIST
|
|
from torchvision import transforms
|
|
from torch.utils.data import DataLoader
|
|
|
|
|
|
mnist_train = MNIST(root="./MNIST_data", train=True, download=True, transform=transforms.PILToTensor())
|
|
|
|
a = DataLoader(mnist_train, batch_size=1, shuffle=True)
|
|
for i in a:
|
|
print(i)
|