ConvertImageDtype

类 torchvision.transforms.ConvertImageDtype(dtype: dtype)

将张量图像转换为给定的 dtype 并相应地缩放其值。

此函数不支持 PIL 图像。

参数：

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

注意

将较小的整数 dtype 转换为较大的整数 dtype 时，最大值不会完全精确地映射。如果来回转换，此不匹配没有影响。

引发：

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

forward(image)

定义每次调用时执行的计算。

应被所有子类重写。

注意

虽然前向传播的实现需要在函数内部定义，但之后应该调用 Module 实例而不是直接调用此函数，因为前者会负责运行已注册的钩子，而后者会默默忽略它们。