关于pytorch的这个错误怎么改?

output with shape [1, 32, 32] doesn't match the broadcast shape [3, 32, 32]这个错怎么改呀,我试过很多解决方法解决不了

代码逻辑问题吧
你看看报错的地方维度为什么不匹配

这个简单, 你在读取图片的时候直接变成灰度图: img = cv.imread(filename, cv.IMREAD_GRAYSCALE)
[1, 32, 32] 是一个channel, [3, 32, 32]是RGB三个channel