ToTensor¶
- 類 torchvision.transforms.v2.ToTensor[source]¶
[已棄用] 請改用
v2.Compose([v2.ToImage(), v2.ToDtype(torch.float32, scale=True)])。將 PIL 影像或 ndarray 轉換為張量並相應地縮放值。
警告
v2.ToTensor已棄用，並將在未來的版本中移除。請改用v2.Compose([v2.ToImage(), v2.ToDtype(torch.float32, scale=True)])。輸出在浮點精度方面是等效的。此轉換不支援 torchscript。
將範圍在 [0, 255] 的 PIL 影像或 numpy.ndarray (H x W x C) 轉換為形狀為 (C x H x W)、範圍在 [0.0, 1.0] 的 torch.FloatTensor，如果 PIL 影像屬於以下模式之一 (L, LA, P, I, F, RGB, YCbCr, RGBA, CMYK, 1) 或者 numpy.ndarray 的 dtype = np.uint8
在其他情況下，張量將不進行縮放地返回。
注意
由於輸入影像被縮放到 [0.0, 1.0]，因此在轉換目標影像掩碼時，不應使用此轉換。請參閱參考資料，瞭解如何實現影像掩碼的轉換。
