峥嵘岁月

本文转自：https://blog.csdn.net/u012436149/article/details/73737299tensorflow 自定义 op本文只是简单的翻译了 https://www.tensorflow.org/extend/adding_an_op 的简单部分，高级部分请移步官网。可能需要新定义 c++ operation 的几种情况：现有的 operation ...

关于在tensorflow(下文称tf)中添加op，tf中文社区中有相应教程：http://www.tensorfly.cn/tfdoc/how_tos/adding_an_op.html 但该教程未提及如何将编写的op加入tensorflow框架。亲测不必编译tensorflow源码，直接通过增量编译C++和CUDA程序可以将op注册在tensorflow中。本文以用CPU实现op为例，记录在...