trans = transforms.Compose([transforms.ToTensor(), transforms.Normalize((0.1307,), (0.3081,))])
1.transforms.Normalize((0.1307,), (0.3081,))为:
归一化操作,其中由于MNIST数据集为单通道图像,那么分别只有一个均值和一个标准差(一般的图像会有三个通道,代表三原色)。
transforms.Normalize(mean,std)的计算公式是:
2.transforms.ToTensor()为:
将0-255的像素转换为0-1的值,对后续的神经网络处理有利
如原始的输入图像矩阵格式为(349, 500, 3),通过转换后(3, 349, 500)
3.transforms.Compose()
管理其中的一系列transforms操作