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。

將範圍在 [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]，因此在轉換目標影像遮罩時不應使用此轉換。請參閱參考資料，以實作影像遮罩的轉換。

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

用於覆寫自訂轉換的方法。

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