Pytorch 學習筆記：單通道圖像的transforms歸一化方法

transforms.Normalize 裏面的數據維數要和圖像數據的通道數保持一致，當數據維數爲1時，數據後面要有逗號。

from torchvision import transforms

transform = transforms.Compose([
        transforms.ToTensor(), # 轉爲Tensor
        transforms.Normalize((0.5,), (0.5,)), # 歸一化
                             ])