1 需求
2 torchvision.datasets
CLASS torchvision.datasets.MNIST(root: Union[str, Path], train: bool = True, transform: Optional[Callable] = None, target_transform: Optional[Callable] = None, download: bool = False)
- root
- train
- transform
- download
MNIST — Torchvision 0.20 documentation
2 torchvision.transforms
Transforming and augmenting images — Torchvision 0.20 documentation