encode_png

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

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

参数：

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

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

返回值：

包含 PNG 文件原始字节的一维 int8 张量。

返回类型：

Tensor[1]