PyTorch 添加 C++ 拓展

参考内容：pytorch添加C++拓展简单实战编写及基本功能测试

文章目录

• 第一步：编写C++内容
• • test.h

    第一步：编写C++内容

    test.h

    #include 
    #include 
    torch::Tensor Test_forward_cpu(const torch::Tensor& inputA, const torch::Tensor& inputB);
    std::vector Test_backward_cpu(const torch::Tensor& gradOutput);