1. training_data = datasets.FashionMNIST(root="data",train=True,download=True,transform=ToTensor())
2. test_data = datasets.FashionMNIST(root="data",train=False,download=True,transform=ToTensor() )
其中root是你放训练集/测试集的路径;train是指定它是训练集还是测试集;download是看你要不要下载这个数据集;transform是指定要对数据做什么变换。
其中的ToTensor()代表用这个函数来处理载入的数据。ToTensor()函数其接受PIL Image或numpy.ndarray格式,功能是:先把HWC改为CHW格式;在转换为float类型;在除以255转为0-1之间。
原文链接:https://blog.csdn.net/czhichao/article/details/115242783