第一次写这种这么长的 readme,介绍了如何在 pytorch 中用 CUDA 写自定义扩展的方法,对搞 nlp 或者 cv 的小伙伴们应该有所帮助,地址在这,how to write cuda extensions in pytorch,求 star^_^,顺便说一句 gboard 挺好用的啊...