TensorFlow for Java

在 TensorFlow.org 上查看 查看 GitHub 存放區

TensorFlow Java 可在任何 JVM 上執行,用於建構、訓練及執行機器學習模型,並含有一系列公用程式和架構,可協助相關領域的數據資料學家和開發人員完成大部分常見的工作。世界各地的小型公司到大型企業經常使用 Java 和其他 JVM 語言 (例如 Scala 或 Kotlin),因此如要大規模採用機器學習技術,TensorFlow 是相當具有策略優勢的選項。

存放區

TensorFlow 的 Java 語言繫結在早期是由主要 TensorFlow 存放區提供,而且只在新版核心程式庫準備發布時 (一年只有少數幾次) 推出。現在所有的 Java 相關程式碼都已移到這個存放區,因此這些程式碼可獨立演進和推出,不受 TensorFlow 正式版推出的影響。此外,大部分的建構工作都已從 Bazel 遷移至大部分 Java 開發人員較熟悉的 Maven。

以下說明存放區的版面配置及其中的各種構件:

  • tensorflow-core

    • 組成 TensorFlow for Java 核心語言繫結的所有構件
    • 目標對象:在 TensorFlow 以外提供自有 API 或架構的專案,只需要透過精簡層從 JVM 存取 TensorFlow 執行階段
  • tensorflow-framework

    • 使用 TensorFlow 建構及訓練類神經網路時的主要 API
    • 目標對象:類神經網路開發人員
  • ndarray

    • 用於執行 N 維資料 I/O 作業的一般公用程式庫
    • 由 TensorFlow 使用,但不必仰賴 TensorFlow 來運作
    • 目標對象:需要導入 Java N 維陣列的開發人員,不限搭配 TensorFlow 使用

通訊

此存放區是由 TensorFlow JVM Special Interest Group (SIG) 維護。只要訂閱 jvm@tensorflow.org 郵寄清單,就能輕鬆加入本群組;您也可以直接傳送提取要求給這個存放區,或提出任何問題。 您還可以參閱 sig-jvm Gitter 頻道