RuntimeError: output with shape [1, 28, 28] doesn‘t match the broadcast shape [3, 28, 28]

**

RuntimeError: output with shape [1, 28, 28] doesn’t match the broadcast shape [3, 28, 28]

**

RuntimeError: output with shape [1, 208, 256] doesn’t match the broadcast shape [3, 208, 256]

pytorch下跑裂缝类代码运行时遇见shape不匹配问题。

原因：图片格式是灰度图只有一个channel，需要变成RGB图才可以

原文链接：https://blog.youkuaiyun.com/weixin_43159148/article/details/88778371