ToTensor¶
- class torchvision.transforms.ToTensor[原始碼]¶
將 PIL 圖像或 ndarray 轉換為張量，並相應地縮放數值。
此轉換不支援 torchscript。
如果 PIL 圖像屬於以下模式之一 (L, LA, P, I, F, RGB, YCbCr, RGBA, CMYK, 1)，或如果 numpy.ndarray 的 dtype = np.uint8，則將範圍 [0, 255] 中的 PIL 圖像或 numpy.ndarray (H x W x C) 轉換為範圍 [0.0, 1.0] 中形狀為 (C x H x W) 的 torch.FloatTensor。
在其他情況下，將傳回未縮放的張量。
注意
由於輸入圖像已縮放為 [0.0, 1.0]，因此在轉換目標圖像遮罩時，不應使用此轉換。請參閱參考文獻，以了解如何實作圖像遮罩的轉換。
