- 博客(2)
- 资源 (2)
- 收藏
- 关注
RSS订阅原创 pytorch学习笔记（十八）：C 语言扩展 pytorch
上篇博文已经介绍了如何通过 继承 Function ，然后使用python 来扩展 pytorch， 本文主要介绍如何通过 cffi 来扩展 pytorch 。官网给出了一个 MyAdd 的 Demo github地址，本文通过 这个 Demo 来搞定如何 通过 cffi 来扩展 pytorch。从github上clone下来代码，目录结构是这样的package：script：（这个部分的示例
2017-12-20 11:22:11
5667
2
原创 pytorch学习笔记（十七）：python 端扩展 pytorch
pytorch 虽然提供了很多的 op 使得我们很容易的使用。但是当已有的 op 无法满足我们的要求的时候，那就需要自己动手来扩展。 pytorch 提供了两种方式来扩展 pytorch 的基础功能。 通过继承 autograd.Function通过 C 来扩展本篇博客主要介绍 继承 autograd.Function 来扩展 pytorch。官方文档链接继承 autograd.Functio
2017-12-18 02:56:07
12841
4
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人