org.tensorflow

Xác định các lớp để xây dựng, lưu, tải và thực thi các mô hình TensorFlow.

CẢNH BÁO : API hiện đang thử nghiệm và không được đảm bảo tính ổn định của API TensorFlow. Xem README.md để biết hướng dẫn cài đặt.

Ví dụ về LabelImage thể hiện việc sử dụng API này để phân loại hình ảnh bằng cách sử dụng mạng thần kinh tích chập kiến ​​trúc Inception được đào tạo trước. Nó thể hiện:

  • Xây dựng biểu đồ: sử dụng lớp OperationBuilder để xây dựng biểu đồ nhằm giải mã, thay đổi kích thước và chuẩn hóa hình ảnh JPEG.
  • Tải mô hình: Sử dụng Graph.importGraphDef() để tải mô hình Khởi động được đào tạo trước.
  • Thực thi biểu đồ: Sử dụng Phiên để thực thi biểu đồ và tìm nhãn tốt nhất cho hình ảnh.

Các ví dụ bổ sung có thể được tìm thấy trong kho lưu trữ tensorflow/models GitHub.

Giao diện

Thực thiMôi trường Xác định môi trường để tạo và thực thi Operation TensorFlow.
Graph.whileSubgraphBuilder Được sử dụng để khởi tạo một lớp trừu tượng ghi đè phương thức buildSubgraph để xây dựng sơ đồ con có điều kiện hoặc nội dung cho vòng lặp while.
Toán hạng <T mở rộng TType > Giao diện được triển khai bởi các toán hạng của hoạt động TensorFlow.
Hoạt động Thực hiện tính toán trên Tensors.
OperationBuilder Trình xây dựng cho Operation s.
Tenxơ Một mảng đa chiều được gõ tĩnh.

Các lớp học

bê tôngChức năng Một biểu đồ có thể được gọi dưới dạng một hàm duy nhất, có chữ ký đầu vào và đầu ra.
Thông số thiết bị Thể hiện thông số kỹ thuật (có thể một phần) cho thiết bị TensorFlow.
DeviceSpec.Builder Lớp Builder để xây dựng lớp DeviceSpec .
Phiên háo hức Một môi trường để thực hiện các hoạt động TensorFlow một cách háo hức.
EagerSession.Options
đồ thị Biểu đồ luồng dữ liệu biểu thị tính toán TensorFlow.
Đồ thịHoạt động Việc triển khai một Operation được thêm dưới dạng nút vào Graph .
Đồ thịHoạt độngXây dựng Một OperationBuilder để thêm GraphOperation vào một Graph .
Đầu ra <T mở rộng TType > Một tay cầm tượng trưng cho một tensor được tạo ra bởi một Operation .
Tensor thô Một tensor mà bộ nhớ chưa được ánh xạ tới không gian dữ liệu có thể truy cập trực tiếp từ JVM.
Đã lưuMô hìnhGói SavingModelBundle đại diện cho một mô hình được tải từ bộ lưu trữ.
Đã lưuModelBundle.Exporter Các tùy chọn để xuất SavingModel.
Đã lưuModelBundle.Loader Các tùy chọn để tải SavingModel.
Máy chủ Máy chủ TensorFlow đang trong quá trình sử dụng để đào tạo phân tán.
Phiên họp Trình điều khiển để thực thi Graph .
Phiên.Run Các tensor đầu ra và siêu dữ liệu thu được khi thực hiện một phiên.
Phiên.Runner Chạy Operation s và đánh giá Tensors .
Chữ ký Mô tả đầu vào và đầu ra của một thực thể có thể thực thi được, chẳng hạn như ConcreteFunction , cùng với các siêu dữ liệu hữu ích khác.
Chữ ký.Builder Xây dựng một chữ ký chức năng mới.
Chữ ký.TensorMô tả
Dòng chảy căng Các phương thức tiện ích tĩnh mô tả thời gian chạy TensorFlow.
TensorMapper <T mở rộng TType > Ánh xạ bộ nhớ riêng của RawTensor tới không gian dữ liệu được nhập n chiều có thể truy cập được từ JVM.

Enum

DeviceSpec.DeviceType
EagerSession.DevicePlacementPolicy Kiểm soát cách hành động khi chúng ta cố gắng chạy một thao tác trên một thiết bị nhất định nhưng một số tensor đầu vào không có trên thiết bị đó.
Thực thiEnvironment.Types