search

godweiyang / godweiyang.github.io

Star & fork my blog source code!
https://github.com/godweiyang/hexo-matery-modified
18 stars 13 forks source link

三分钟教你如何PyTorch自定义反向传播 | 韦阳的博客 #237

Open godweiyang opened 3 years ago

godweiyang commented 3 years ago

https://godweiyang.com/2021/03/24/torch-cpp-cuda-3/

在前面两篇教程中，我们详细讲解了如何编写cuda算子，并用PyTorch进行调用，并且详细讲述了三种编译cuda算子的方式，具体可以看前面两篇： https://godweiyang.com/2021/03/18/torch-cpp-cud