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<torch::Tensor> Test_backward_cpu(const torch::Tensor& gradOutput);

你可能感兴趣的:(PyTorch,C++,pytorch,linux,c++)