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 例項而不是直接呼叫此函式，因為前者會負責執行已註冊的鉤子，而後者會默默忽略它們。