转换图像数据类型

class torchvision.transforms.ConvertImageDtype(dtype: dtype)

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

此函数不支持PIL图像。

Parameters:

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

注意

当从较小的整数转换为较大的整数dtype时，最大值不会精确映射。 如果来回转换，这种不匹配没有影响。

Raises:

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

forward(image)

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

应该由所有子类覆盖。

注意

尽管前向传递的配方需要在此函数内定义，但之后应该调用Module实例而不是这个，因为前者负责运行已注册的钩子，而后者则默默地忽略它们。