encode_jpeg

torchvision.io.encode_jpeg(input: Union[Tensor, list[torch.Tensor]], quality: int = 75) → Union[Tensor, list[torch.Tensor]]

在 CPU 或 CUDA 上將 RGB 張量編碼為原始編碼的 jpeg 位元組。

注意

傳遞 CUDA 張量列表比重複單獨呼叫 encode_jpeg 更有效。對於 CPU 張量，效能相當。

引數:

• input (Tensor[通道, 影像高度, 影像寬度] 或 List[Tensor[通道, 影像高度, 影像寬度]]) – 一個或多個 c 通道的 uint8 影像張量，其中 c 必須是 1 或 3。

• quality (int) – 生成的 JPEG 檔案(s)的質量。必須是介於 1 和 100 之間的數字。預設為：75

返回:

一個或多個包含 JPEG 檔案原始位元組的一維 uint8 張量。

返回型別:

output (Tensor[1] 或 list[Tensor[1]])