提交 2a4b9c7c 编写于 作者: Z ZhanPw

Update cpp_extension.md

上级 5ec9925f
......@@ -2,6 +2,8 @@
> 译者:[P3n9W31](https://github.com/P3n9W31)
**Author**: [Peter Goldsborough](https://www.goldsborough.me/)
PyTorch 提供了大量与神经网络,任意张量代数(arbitrary tensor algebra),数据处理(data wrangling)和其他目的相关的操作。然而,你可能发现你还是会需要一些更加自定义的操作。例如,你有时可能希望使用一个你在某篇论文中找到的一个新型的激活函数,或者是实现一个为了你的研究所开发的新操作。
在 PyTorch 中集成这种自定义操作的最简单方法是通过 Python 语言对`Function``Module`进行扩写,正如在 [这里](https://pytorch.org/docs/master/notes/extending.html)所描述的那样。这种方式能让你充分的发挥自动微分(automatic differentiation)(让你不用去编写一些衍生的函数)与 Python 语言的常规情况下的表现力(usual expressiveness)的能力。但是有时候,可能在 C++ 语言中能够更好地实现你的一些操作。例如,你的代码可能因为被非常频繁的使用而需要 _十分_ 快速,或者是即使调用的次数很少也会带来不小的性能负担。另一个原因是你的代码可能是建立在 C 或 C++ 语言之上的,或者你的代码需要与 C 或 C++ 语言进行交互与对接。为了解决上述的这些情况,PyTorch 提供了一种简单的用于编写自定义 _C++ 扩展_ 的方法。
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册