c++ - TensorFlow C 中的自定义操作
问题描述
我正在尝试使用 Tensorflow C API 创建一个 TF 自定义操作。我已经阅读了为 C添加新操作并安装 TensorFlow，但几乎没有使用 Tensorflow C 创建自定义操作的教程。所以，我想知道是否有人可以给我看一个小示例代码？
我不能在我的环境中使用 bazel 构建系统，所以我假设我不能使用 Tensorflow C++ API，因为我只需要为自定义操作代码使用静态或共享库。请让我知道在 Tensorflow C++ API 中是否仍然可以执行此任务。
解决方案
回答您的第二个问题：此任务在 TensorFlow C++ API 中是可能的，但要构建 TensorFlow C++ API，您确实需要 bazel。