#pragma once #include "readjpeg_cpu.h" #include "readpng_cpu.h"
C10_EXPORT torch::Tensor decode_image( const torch::Tensor& data, int64_t channels = 0);