encode_png¶
- torchvision.io.encode_png(input: Tensor, compression_level: int = 6) Tensor[原始碼]¶
接受一個採用 CHW 佈局的輸入張量，並返回一個包含其對應 PNG 檔案內容的緩衝區。
- 引數:
input (Tensor[channels, image_height, image_width]) – 具有
c個通道的 int8 影像張量，其中c必須為 3 或 1。compression_level (int) – 結果檔案的壓縮因子，必須是介於 0 和 9 之間的數字。預設為 6。
- 返回:
- 一個一維 int8 張量，包含原始位元組資料，即
PNG 檔案。
- 返回型別:
Tensor[1]
