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 實例，而不是這個函式，因為前者會處理執行註冊的 hook，而後者會靜默地忽略它們。