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]
