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 图像属于模式（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] 中的 torch.FloatTensor，形状为 (C x H x W)。

在其他情况下，将返回未缩放的张量。

注意

由于输入图像被缩放到 [0.0, 1.0]，因此在转换目标图像掩码时，不应使用此转换。请参阅 参考文献，了解如何实现图像掩码的转换。

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

用于覆盖自定义转换的方法。

请参阅 如何编写您自己的 v2 转换