tensorflow - 使用 libtensorflow_cc.so 使用 Tensorflow C++ API 的 header 列表

Question

我想知道使用 Tensorflow 的 C++ API 需要哪些头文件。与 C API 的情况一样，只有一个 header c_api.h，其中声明了所有函数等，C++ API 是否有这样的单个 header ？我尝试搜索此内容，但无法理解需要什么和不需要什么。

在tensorflow/cc、tensorflow/core和tensorflow/c中有一个巨大的头文件列表，用于构建libtensorflow_cc.so，我们也在tensorflow的发行版中提供了其中的大部分(我所说的TF发行版是指TF，它是使用 bazel build -c opt --config=cuda //tensorflow/tools/pip_package:build_pip_package 构建)。该 header 列表是否足以使用 C++ API？或者我们是否需要在tensorflow/BUILD中构建任何额外的目标？

我也浏览过https://www.tensorflow.org/api_docs/cc/但无法真正确定所需 header 的确切列表。

在一篇相关文章中，我发现tensorflow/bazel-genfiles包含所需的 header 。请确认这一点。

提前致谢，西达

Answer 1

对于使用 bazel 构建适用于 Windows 的 TensorFlow v2 C++ 的用户，请使用

bazel build --config=opt tensorflow:install_headers

这将在 bazel-bin/tensorflow/ 中生成一个 include 文件夹，其中包含 cc 头文件。