今天小编就为大家分享一篇TensorFlow实现自定义Op方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

『写在前面』 以CTC Beam search decoder为例，简单整理一下TensorFlow实现自定义Op的操作流程。 基本的流程 1. 定义Op接口 #include "tensorflow/core/framework/op.h" REGISTER_OP("Custom") .Input("custom_input: int32") .Output("custom_output: int32"); 2. 为Op实现Compute操作(CPU)或实现kernel(GPU)