[PyTorch] output with shape [1, 28, 28] doesn’t match the broadcast shape [3, 28, 28] 4 6 月, 2021 會產生這一個錯誤,原因是圖片的格式造成,是彩色還…