快捷方式

ToTensor

class torchvision.transforms.v2.ToTensor[源代码]

[已弃用] 请改用 v2.Compose([v2.ToImage(), v2.ToDtype(torch.float32, scale=True)])

将 PIL Image 或 ndarray 转换为张量并相应地缩放值。

警告

v2.ToTensor 已弃用,将在未来版本中移除。请改用 v2.Compose([v2.ToImage(), v2.ToDtype(torch.float32, scale=True)])。输出在浮点精度上是等价的。

此转换不支持 torchscript。

将范围为 [0, 255] 的 PIL Image 或 numpy.ndarray (H x W x C) 转换为范围为 [0.0, 1.0] 的形状为 (C x H x W) 的 torch.FloatTensor,如果 PIL Image 属于以下模式之一 (L, LA, P, I, F, RGB, YCbCr, RGBA, CMYK, 1) 或如果 numpy.ndarray 的 dtype = np.uint8。

在其他情况下,张量将不进行缩放返回。

注意

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

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发人员的深度教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源