readpng_cpu.h 163 Bytes
Newer Older
1
2
3
4
5
6
#pragma once

// Comment
#include <torch/torch.h>
#include <string>

7
8
9
C10_EXPORT torch::Tensor decodePNG(
    const torch::Tensor& data,
    int64_t channels = 0);