Pytorch使用Python,CPP,CUDA进行拓展
使用CPP 和CUDA进行扩展时面临着几个小问题:python与cpp的接口绑定方式, python与cpp的数据交互方式,cpp编译工具的选择。现在逐条回答:
数据接口
ATen: The foundational tensor and mathematical operation library on which all else is built;
ATen
ATen is fundame...
原创
2019-07-04 10:49:00 ·
1378 阅读 ·
0 评论