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