encode_jpeg

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

将 RGB 张量编码为原始编码的 JPEG 字节，可在 CPU 或 CUDA 上进行。

注意

传递 CUDA 张量列表比重复单独调用 encode_jpeg 更高效。对于 CPU 张量，性能是相同的。

参数:

• input (Tensor[通道数, 图像高度, 图像宽度] 或 List[Tensor[通道数, 图像高度, 图像宽度]]) – (列表) uint8 图像张量，具有 c 个通道，其中 c 必须为 1 或 3

• quality (int) – 生成的 JPEG 文件(s)的质量。必须是介于 1 和 100 之间的数字。默认值：75

返回:

一个（列表）一维 uint8 张量，其中包含 JPEG 文件的原始字节。

返回类型:

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