在tensorflow中编写自定义层:找不到op_def_builder.h没有这样的文件或目录

时间:2018-01-20 15:30:28

标签: python c++ tensorflow

我通过tensorflow安装了pip,当我尝试在cpp中编写自定义图层时,我找不到所需的.h文件。例如 tensorflow/core/framework/op_def.pb.h我做了一些研究并得出结论,我需要克隆github代码并构建库来生成该文件。这是真的?如果这是唯一的方法,为什么tensorflow docs建议我可以通过安装tensorflow二进制文件或从源代码构建库来实现此目的?

注意:tensorflow/core/framework/op_def_builder.h需要#include tensorflow/core/framework/op_def_builder.h#include "tensorflow/core/framework/op.h"包含github,我可以在~

上发布的代码中找到+

0 个答案:

没有答案