encode_png¶
- torchvision.io.encode_png(input: Tensor, compression_level: int = 6) Tensor[源代码]¶
接受一个采用 CHW 布局的输入张量，并返回一个包含其对应 PNG 文件内容的缓冲区。
- 参数:
input (Tensor[channels, image_height, image_width]) – 具有
c个通道的 int8 图像张量，其中c必须为 3 或 1。compression_level (int) – 结果文件的压缩因子，必须是介于 0 和 9 之间的数字。默认为 6。
- 返回:
- 一个一维 int8 张量，包含原始字节数据，即
PNG 文件。
- 返回类型:
Tensor[1]
