RuntimeError: output with shape [1, 28, 28] doesn't match the broadcast shape [3, 28, 28]
将transforms.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5])
改为:transforms.Normalize([0.5],[0.5])
参考文章:https://blog.csdn.net/weixin_43159148/article/details/88778371#commentsedit