ConvertImageDtype

class torchvision.transforms.ConvertImageDtype(dtype: dtype)

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

此函数不支持 PIL Image。

参数：

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

注意

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

引发：

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

forward(image)

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

应由所有子类覆盖。

注意

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