TensorFlow for Java

在 TensorFlow.org 上查看 查看 GitHub 代码库

TensorFlow Java 可在任何 JVM 上运行，用于构建、训练和运行机器学习模型。它附带一系列实用程序和框架，可帮助这一领域的数据科学家和开发者完成大多数常见任务。世界各地的大小企业都在频繁使用 Java 和其他 JVM 语言（比如 Scala 或 Kotlin），有鉴于此，若要大规模采用机器学习，TensorFlow 无疑是一项战略选择。

代码库

早期，TensorFlow 的 Java 语言绑定托管在 TensorFlow 主代码库中，只有当核心库的新版本做好分发准备后，这些绑定才会发布，而这种情况每年只发生几次。现在，所有与 Java 相关的代码都已迁移到该代码库，因此，这些代码可独立于官方 TensorFlow 版本进行更新和发布。此外，大多数构建任务也已从 Bazel 迁移到 Maven，大多数 Java 开发者对后者更为熟悉。

以下内容介绍了该代码库的布局及其不同的工件：

  • tensorflow-core

    • 用于构建 TensorFlow for Java 的核心语言绑定的所有工件
    • 目标受众：提供自己的基于 TensorFlow 的 API 或框架，并且只想使用薄层通过 JVM 访问 TensorFlow 运行时的项目

  • tensorflow-framework

    • 用于使用 TensorFlow 构建和训练神经网络的主 API
    • 目标受众：神经网络开发者

  • ndarray

    • 用于 N 维数据 I/O 操作的通用实用程序库
    • 由 TensorFlow 使用，但不依赖于 TensorFlow
    • 目标受众：任何需要 Java N 维数组实现的开发者，无论是否将其与 TensorFlow 结合使用

沟通

该库由 TensorFlow JVM 特别兴趣小组 (SIG) 进行维护。您只需订阅 jvm@tensorflow.org 网上论坛即可轻松地加入该小组，或者，您也可以直接向该代码库发送拉取请求和提出问题。此外，我们还提供了一个 sig-jvm Gitter 渠道。