encode_png

torchvision.io.encode_png(input: Tensor, compression_level: int = 6) → Tensor

接受 CHW 布局的输入张量，并返回包含其对应 PNG 文件内容的缓冲区。

参数:

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

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

返回:

包含原始字节的一维 int8 张量

PNG 文件。

返回类型:

Tensor[1]