Pytorch常见问题（一）：MNIST集output with shape [1, 28, 28] doesn't match the broadcast shape [3, 28, 28]

主要原因：MNIST集中的图片为灰度图片，只有一个通道，而

transform = transforms.Compose([
transforms.ToTensor(),
transforms.Normalize(mean = (0.5,0.5,0.5),
std = (0.5,0.5,0.5))])

代码中mean=(0.5,0.5,0.5)则为三个通道的均值，故不匹配。
【但是大多数学习Pytorch的书里代码都是这么写的，遇到类似问题的人也很多，不解。是版本问题吗？】

修改方法1：

方法及原因都讲解的比较清楚了

修改方法2：

transform = transforms.Compose([
transforms.ToTensor(),
transforms.Lambda(lambda x: x.repeat(3,1,1)), #添加这行
transforms.Normalize(mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5))
])

将图片通道修改为3，但是这种修改方法感觉比较暴力，后续容易出现问题（之后所有跟通道有关的参数都要改）