ConvertImageDtype

class 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 实例，而不是此函数，因为前者负责运行已注册的钩子，而后者会默默忽略它们。