ToTensor

class torchvision.transforms.v2.ToTensor

[已棄用] 請改用 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。

將 PIL Image 或 numpy.ndarray (H x W x C) 範圍在 [0, 255] 轉換為形狀為 (C x H x W) 範圍在 [0.0, 1.0] 的 torch.FloatTensor，如果 PIL Image 屬於以下模式之一 (L, LA, P, I, F, RGB, YCbCr, RGBA, CMYK, 1) 或 numpy.ndarray 的 dtype = np.uint8。

在其他情況下，張量不進行縮放直接返回。

注意

由於輸入影像被縮放到 [0.0, 1.0]，因此在轉換目標影像掩碼時，不應使用此轉換。有關實現影像掩碼轉換的方法，請參閱 參考資料。

transform(inpt: Union[Image, ndarray], params: dict[str, Any]) → Tensor

用於覆蓋自定義變換的方法。

請參閱 如何編寫自己的 v2 變換