ToTensor¶
- class torchvision.transforms.ToTensor[原始碼]¶
將 PIL 影像或 ndarray 轉換為張量並相應地縮放值。
此轉換不支援 torchscript。
將 PIL 影像或 numpy.ndarray (H x W x C)（範圍在 [0, 255]）轉換為 torch.FloatTensor（形狀為 (C x H x W)，範圍在 [0.0, 1.0]），如果 PIL 影像屬於模式（L, LA, P, I, F, RGB, YCbCr, RGBA, CMYK, 1）之一，或者 numpy.ndarray 的 dtype 為 np.uint8。
在其他情況下，張量不進行縮放直接返回。
注意
由於輸入影像被縮放到 [0.0, 1.0]，因此在轉換目標影像掩碼時，不應使用此變換。有關實現影像掩碼變換的方法，請參閱參考資料。
