Pytorch学习(一)------自定义C++/ATen扩展

具体的详细细节可参考此篇博客，非常的细了

作用本质

本质其实就是用C++代码写的功能封装成python可以import的功能包，可以执行的更快！！

环境要求

老老实实用python3.7，高版本编译总会有想不到的bug
手动去这个网址找适合你的版本下载安装！！！

使用JIT Compile

对应着静态编译的setup.py，我们新建一个编译脚本jit.py

from torch.utils.cpp_extension import load

lltm_cpp = load(name="lltm_cpp", sources=["lltm.cpp"], verbose=True)