Question

最近，我正在尝试学习如何使用Mac OS Xcode在tensorflow中添加新操作。但是，我找不到tensorflow头文件，并且我已经用pip安装了tensorlfow。我不确定如何解决这个问题。以下代码在tensorlfow文档中是公开的。

#include "tensorflow/core/framework/op.h"
#include "tensorflow/core/framework/shape_inference.h"

using namespace tensorflow;

REGISTER_OP("ZeroOut")
    .Input("to_zero: int32")
    .Output("zeroed: int32")
    .SetShapeFn([](::tensorflow::shape_inference::InferenceContext* c) {
      c->set_output(0, c->input(0));
      return Status::OK();
    });

我只想重复一遍，但是我找不到tensorflow的路径。非常感谢大家！

Answer 1

您要么使用bazel在TensorFlow存储库中工作，要么需要编写自己的Makefile，而仅使用pip包。

如果您只是为TensorFlow创建操作，而您正在使用v1.9，则可以使用以下方式从pip包中查找这些信息

import tensorflow as tf
print(tf.sysconfig.get_include())
print(tf.sysconfig.get_lib()

但是您可能需要much more information来编译代码。

如何使用Xcode在Tensorflow中添加操作

1 个答案: