java - java、Tensorflow中的全连接层

Question

如何在Java中使用Tensorflow制作全连接层？

从他们的“Keynote (TensorFlow Dev Summit 2017)”演讲者那里，我得到的印象是你可以用 JAVA 来对 Tensorflow 进行编程。

来自他们的documentation我无法找到它的支持。该文档让我相信我只能导入模型。

Answer 1

TensorFlow Java API 最适合导入和执行模型。它确实支持构建 TensorFlow 操作图，但没有 Python 具有的高级便利 API(层库、估计器类、优化器等)。因此，一般来说，我建议使用 Python 构建图表，然后将它们导入到您的 Java 应用程序中。

也就是说，如果您确实想构建图表，请从标签图像样本 builds a "model" for normalizing the image 中获取灵感。 。

完全连接的层将涉及矩阵乘法、偏置添加和激活函数，因此您必须将这些操作添加到图中。不幸的是，目前弄清楚这一切并不容易。 https://github.com/tensorflow/tensorflow/issues/7149应该会有所帮助 - 因为至少会为所有 TensorFlow 操作生成代码和文档。不过，这个问题还没有解决。

希望有帮助。