转换为张量¶

class 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]，所以在转换目标图像掩码时不应使用此转换。请参阅references以了解如何实现图像掩码的转换。