1 导入MNIST 和UPSP数据集：
从torchvision.datasets.MNIST下载即可，代码在datasets文件夹里mnist.py 和usps.py
下载后的MNIST图像都是灰度图像，只有一个通道。所以运行原来的程序会报错：
RuntimeError: output with shape [1, 28, 28] doesn't match the broadcast shape [3, 28, 28]。
原来程序：
原来程序
报错原因：这是因为mnist图像都是灰度图像，只有一个通道，而上面的transforms.Normalize 却对三个通道都归一化了，这肯定会报错，所以只要像下面修改即可：
修改后
