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 例項而不是此函式，因為前者會處理註冊的鉤子，而後者則會靜默忽略它們。