2019年6月2日
PyTorch中的C++扩展
摘要： 今天要聊聊用 PyTorch 进行 C++ 扩展。 在正式开始前，我们需要了解 PyTorch 如何自定义module。这其中，最常见的就是在 python 中继承torch.nn.Module，用 PyTorch 中已有的 operator 来组装成自己的模块。这种方式实现简单，但是，计算效率却未 阅读全文
posted @ 2019-06-02 14:57 大白话AI 阅读(10373) 评论(0) 推荐(3) 编辑