tensorflow/java

Java 版 TensorFlow

是什麼 – 一組 Java 語言綁定，讓您可在任何 JVM 上執行 TensorFlow 模型。此倉儲包含三個主要元件：

• tensorflow-core – 對原生 TensorFlow 庫的輕量級 JNI 包裝。若您僅需低階存取，請使用此元件。
• tensorflow-framework – 用於在 Java 中直接建構與訓練神經網路的高階 API。
• tensorflow-ndarray – 一個獨立的 n 維陣列函式庫，可脫離 TensorFlow 其他部分使用。

為何重要 – Java、Scala、Kotlin 等 JVM 語言在大型企業中相當常見。透過提供 Maven 風格的元件，TensorFlow-Java 讓這些團隊能在不離開既有建構與部署流程的情況下採用機器學習。

快速入門 – 加入適當的 Maven（或 Gradle）相依性。您可以選擇：

1. 個別挑選原生二進位檔（tensorflow-core-api + tensorflow-core-native，並附加平台分類器），以保持 JAR 檔案較小；或
2. 使用單一的 tensorflow-core-platform 元件，該元件捆綁所有支援的 CPU 平台（Linux x86_64、Linux arm64、macOS arm64）的 API 與原生程式庫。

範例（Maven，Linux-x86_64，僅 CPU）：

<dependency>
  <groupId>org.tensorflow</groupId>
  <artifactId>tensorflow-core-api</artifactId>
  <version>1.2.0</version>
</dependency>
<dependency>
  <groupId>org.tensorflow</groupId>
  <artifactId>tensorflow-core-native</artifactId>
  <version>1.2.0</version>
  <classifier>linux-x86_64</classifier>
</dependency>

如需 GPU 支援，請加入 linux-x86_64-gpu 分類器，並確保已安裝 NVIDIA 驅動、CUDA 工具包與 cuDNN。

版本相容性 – README 中的表格將每個 TensorFlow-Java 版本對應到底層 TensorFlow 版本與所需的最低 Java 版本（例如，TF-Java 1.2.0 與 TensorFlow 2.21.0 相容，需 Java 11）。

社群與貢獻 – 由 TensorFlow JVM SIG 維護。您可在 TensorFlow 論壇（sig_jvm 標籤）或 Gitter 頻道提問，貢獻遵循 CONTRIBUTING.md 中描述的標準拉取請求工作流程。

實際應用範例 – 範例模型位於附屬倉儲 https://github.com/tensorflow/java-models 中。

---

• 上述所有細節均直接取自倉儲的 README。*