write_png Tensor, filename: str, compression_level: int = 6)[source]

Takes an input tensor in CHW layout (or HW in the case of grayscale images) and saves it in a PNG file.

  • input (Tensor[channels, image_height, image_width]) – int8 image tensor of c channels, where c must be 1 or 3.

  • filename (str or pathlib.Path) – Path to save the image.

  • compression_level (int) – Compression factor for the resulting file, it must be a number between 0 and 9. Default: 6


Access comprehensive developer documentation for PyTorch

View Docs


Get in-depth tutorials for beginners and advanced developers

View Tutorials


Find development resources and get your questions answered

View Resources