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