python報錯：1only batches of spatial targets supported (non-empty 3D tensors) but got targets of size

因為在使用Crossentropyloss作為損失函數時，output=net(input)的output應該是[batchsize, channel, height, weight]，而label則是[batchsize, height, weight]，label是單通道灰度圖，而在BCELoss中，兩者都是[batchsize, channel, height, weight]。

我報錯是因為label本應該是一維的，我在保存的時候處理成了3維。