write_jpeg

torchvision.io.write_jpeg(input: Tensor, filename: str, quality: int = 75)

接受 CHW 布局的输入张量并将其保存为 JPEG 文件。

参数：

• input (Tensor[通道数, 图像高度, 图像宽度]) – int8 图像张量，包含 c 个通道，其中 c 必须为 1 或 3。

• filename (str 或 pathlib.Path) – 保存图像的路径。

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