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