#pragma once

#include <torch/script.h>
#include <torch/torch.h>
#include "readpng_cpu.h"