encode_png¶
-
torchvision.io.
encode_png
(input: torch.Tensor, compression_level: int = 6) → torch.Tensor[source]¶ Takes an input tensor in CHW layout and returns a buffer with the contents of its corresponding PNG file.
- Parameters
input (Tensor[channels, image_height, image_width]) – int8 image tensor of
c
channels, wherec
must 3 or 1.compression_level (int) – Compression factor for the resulting file, it must be a number between 0 and 9. Default: 6
- Returns
- A one dimensional int8 tensor that contains the raw bytes of the
PNG file.
- Return type
Tensor[1]