torch-ext/torch_binding.h

#include <torch/torch.h>

void img2gray_cuda(torch::Tensor input, torch::Tensor output);