分享

怎样用C 写出你自己的TensorFlow?

 山峰云绕 2018-07-26

Tensorflow是一款功能强大且设计精良的神经网络工具。 Python API已有详细记录,开始非常简单。 另一方面,C ++ API的文档减少到最低限度。 本教程将向您展示如何:

  • 使用Python构建和训练一个简单的图形,

  • 保存图形并在C ++中运行它。

在本教程中,我们将使用Google自己的构建工具bazel。 如果您更喜欢在没有bazel的情况下工作,请查看如何让Tensorflow在没有bazel的情况下运行。 作为一个例子,我们将使用世界上最小的网络。 它只包含一个输入神经元和一个输出神经元。 网络如下所示:

结果目标是为输出和输入获得相同的值。 虽然没有任何意义,但它只是一个例子。 损失函数将是平方误差,要求安装Bazel和Tensorflow。

创建一个文件夹且包含两个文件:

  • 实际的C ++文件RunGraph.cpp;

  • 一个名为BUILD的文件,带有Bazel的说明。

这两个文件都可以从我们的GitHub Repo下载, 我们来看看RunGraph.cpp文件。

首先,您需要使用计算图的路径更改PathGraph。 另一件事是我们不需要提供两个占位符。 我们指定了输出Y.第二个占位符(Y_)仅用于计算损失而不是Y。如果我们将损失指定为输出,程序也将要求第二个占位符,我们来看看BUILD文件。

然后运行以下命令:


希望本教程对您有点帮助。我会在这里发布所有与科技、科学有关的有趣文章,欢迎订阅我的头条号。偶尔也回答有趣的问题,有问题可随时在评论区回复和讨论。

(码字不易,若文章对你帮助可点赞支持~)

    本站是提供个人知识管理的网络存储空间,所有内容均由用户发布,不代表本站观点。请注意甄别内容中的联系方式、诱导购买等信息,谨防诈骗。如发现有害或侵权内容,请点击一键举报。
    转藏 分享 献花(0

    0条评论

    发表

    请遵守用户 评论公约

    类似文章 更多