python cpp cuda混合编程_pytorch通过torch.utils.cpp_extension构建CUDA/C++拓展
torch.utils.cpp_extension.CppExtension(name,sources,*args,**kwargs)创建一个C++的setuptools.Extension。便捷地创建一个setuptools.Extension具有最小(但通常是足够)的参数来构建C++扩展的方法。所有参数都被转发给setuptools.Extension构造函数。例>>>fromsetuptoo