我真的很渴望开始在 C++ 中使用 Google 新的 Tensorflow 库。网站和文档对于如何构建项目的 C++ API 确实不清楚,我不知道从哪里开始。
有更多经验的人可以通过发现和分享使用 TensorFlow 的 C++ API 的指南来提供帮助吗?
首先,您应该从 Github 下载源代码,方法是按照此处的说明进行操作(你需要Bazel和最新版本的 GCC)。
C++ API(以及系统的后端)位于tensorflow/core
。目前,只有C++ 会话接口,以及C API正在受到支持。您可以使用其中任何一个来执行使用 Python API 构建并序列化为GraphDef
协议缓冲区。还有一个用于在 C++ 中构建图的实验性功能,但目前它的功能还不如 Python API 功能齐全(例如,目前不支持自动微分)。您可以看到一个示例程序在这里用 C++ 构建一个小图.
C++ API 的第二部分是添加新的 APIOpKernel
,这是包含 CPU 和 GPU 数值内核实现的类。有很多关于如何构建这些的示例tensorflow/core/kernels
, 以及ASA在 C++ 中添加新操作的教程.
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)