write_png

torchvision.io.write_png(input: Tensor, filename: str, compression_level: int = 6)

將 CHW 佈局（或灰度影像的 HW）的輸入張量儲存為 PNG 檔案。

引數:

• input (Tensor[channels, image_height, image_width]) – int8 影像張量，具有 c 個通道，其中 c 必須為 1 或 3。

• filename (str 或 pathlib.Path) – 要儲存影像的路徑。

• compression_level (int) – 輸出檔案的壓縮因子，它必須是一個介於 0 和 9 之間的數字。預設值：6