第一次写这种这么长的 readme，介绍了如何在 pytorch 中用 CUDA 写自定义扩展的方法，对搞 nlp 或者 cv 的小伙伴们应该有所帮助，地址在这，how to write cuda extensions in pytorch，求 star^_^，顺便说一句 gboard 挺好用的啊...
第一次写这种这么长的 readme，介绍了如何在 pytorch 中用 CUDA 写自定义扩展的方法，对搞 nlp 或者 cv 的小伙伴们应该有所帮助，地址在这，how to write cuda extensions in pytorch，求 star^_^，顺便说一句 gboard 挺好用的啊...
 |
1
zhuangjia Mar 16, 2018
再提供一个中文版可能 star 会更多点？
|
Select Language