快捷方式

ConvertImageDtype

class torchvision.transforms.v2.ConvertImageDtype(dtype: dtype = torch.float32)[源代码]

[已弃用] 请改用 v2.ToDtype(dtype, scale=True)

将输入图像转换为给定的 dtype 并相应地缩放值。

警告

请考虑改用 ToDtype(dtype, scale=True)。请参阅 ToDtype

此函数不支持 PIL Image。

参数:

dtype (torch.dpython:type) – 输出所需的数据类型

注意

从较小的整数 dtype 转换为较大的整数 dtype 时,最大值**不会**完全映射。如果来回转换,此不匹配将不会产生影响。

引发:

RuntimeError – 当尝试将 torch.float32 转换为 torch.int32torch.int64 以及当尝试将 torch.float64 转换为 torch.int64 时。这些转换可能会导致溢出错误,因为浮点数 dtype 无法存储整数 dtype 的整个范围内的连续整数。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

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

查看资源