write_png

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

接受一个CHW布局的输入张量（或在灰度图像的情况下为HW布局）并将其保存为PNG文件。

Parameters:

• 输入 (张量[通道数, 图像高度, 图像宽度]) – 一个c通道的int8图像张量，其中c必须为1或3。

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

• compression_level (int) – 结果文件的压缩级别，必须是一个介于0和9之间的数字。默认值：6