Solved by dashesy in post #4 
              
              
                This is the solution 
    std::vector<at::Tensor> forward(at::Tensor xy, int height, int width) {
        auto tmp  = at::empty({10, 4, height, width}, xy.options().dtype(at::kInt));
        return {tmp}
    }