write_png

torchvision.io.write_png(input: Tensor, filename: str, compression_level: int = 6)

接受 CHW 布局（或灰度图像的 HW）中的输入张量，并将其保存到 PNG 文件中。

参数：:

• input (Tensor[channels, image_height, image_width]) – c 个通道的 int8 图像张量，其中 c 必须是 1 或 3。

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

• compression_level (int) – 生成的文件的压缩因子，必须是 0 到 9 之间的数字。默认值：6