【发布时间】:2019-09-04 00:49:03
【问题描述】:
我正在尝试使用 Tensorflow C API 创建一个 TF 自定义操作。我已经阅读了Adding a New Op 和Install TensorFlow for C,但是几乎没有使用 Tensorflow C 创建自定义操作的教程。所以,我想知道是否有人可以给我看一个小示例代码?
我不能在我的环境中使用 bazel 构建系统,所以我假设我不能使用 Tensorflow C++ API,因为我只需要为自定义操作代码使用静态或共享库。请让我知道在 Tensorflow C++ API 中是否仍然可以执行此任务。
【问题讨论】:
-
@Fureeish 因为 TF 是用 C++ 编写的,所以我认为 C++ 中可能还有一种解决方法,可以不使用 bazel 来解决这个问题!
标签: c++ c tensorflow