丁丁的博客

原创 pytorch 学习笔记之编写 C 扩展

注，pytorch 0.4 提供了更加方便的工具来添加 C++ 扩展，参见这里及这里。pytorch利用 CFFI 进行 C 语言扩展。包括两个基本的步骤（docs）：编写 C 代码； python 调用 C 代码，实现相应的 Function 或 Module。在之前的文章中，我们已经了解了如何自定义 Module。至于 [py]torch 的 C 代码库的结构，我们留...