org.tensorflow

Определяет классы для создания, сохранения, загрузки и выполнения моделей TensorFlow.

ВНИМАНИЕ : API в настоящее время является экспериментальным и не покрывается гарантиями стабильности API TensorFlow. Инструкции по установке см. в README.md .

Пример LabelImage демонстрирует использование этого API для классификации изображений с использованием предварительно обученной сверточной нейронной сети архитектуры Inception . Он демонстрирует:

  • Построение графа: использование класса OperationBuilder для построения графа для декодирования, изменения размера и нормализации изображения JPEG.
  • Загрузка модели: использование Graph.importGraphDef() для загрузки предварительно обученной начальной модели.
  • Выполнение графика: использование сеанса для выполнения графиков и поиска лучшей метки для изображения.

Дополнительные примеры можно найти в репозитории tensorflow/models на GitHub.

Интерфейсы

среда выполнения Определяет среду для создания и выполнения Operation TensorFlow.
Graph.WhileSubgraphBuilder Используется для создания экземпляра абстрактного класса, который переопределяет метод buildSubgraph для построения условного или основного подграфа для цикла while.
Операнд <T расширяет TType > Интерфейс, реализованный операндами операции TensorFlow.
Операция Выполняет вычисления на тензорах.
ОперацияBuilder Строитель для Operation s.
Тензор Статически типизированный многомерный массив.

Классы

КонкретнаяФункция График, который можно вызвать как одну функцию с входной и выходной сигнатурой.
Спецификация устройства Представляет (возможно, частичную) спецификацию устройства TensorFlow.
DeviceSpec.Builder Класс Builder для создания класса DeviceSpec .
Энергичная сессия Среда для быстрого выполнения операций TensorFlow.
EagerSession.Options
График Граф потока данных, представляющий вычисление TensorFlow.
ГрафикОперация Реализация Operation , добавленной в качестве узла в Graph .
ГрафОперацияBuilder OperationBuilder для добавления GraphOperation в Graph .
Вывод <T расширяет TType > Символический дескриптор тензора, созданного Operation .
RawTensor Тензор, память которого не сопоставлена ​​с пространством данных, напрямую доступным из JVM.
СохраненнаяМодельБундле SavedModelBundle представляет модель, загруженную из хранилища.
SavedModelBundle.Exporter Параметры экспорта SavedModel.
SavedModelBundle.Loader Параметры загрузки SavedModel.
Сервер Внутренний сервер TensorFlow для использования в распределенном обучении.
Сессия Драйвер для выполнения Graph .
Сессия.Выполнить Выходные тензоры и метаданные, полученные при выполнении сеанса.
Сессия.Бегун Запустите Operation и оцените Tensors .
Подпись Опишите входные и выходные данные исполняемого объекта, такого как ConcreteFunction , а также другие полезные метаданные.
Подпись.Строитель Создает новую сигнатуру функции.
Signature.TensorDescription
Тензорфлоу Статические служебные методы, описывающие среду выполнения TensorFlow.
TensorMapper <T расширяет TType > Сопоставляет собственную память RawTensor с n-мерным пространством типизированных данных, доступным из JVM.

Перечисления

DeviceSpec.DeviceType
EagerSession.DevicePlacementPolicy Управляет тем, как действовать, когда мы пытаемся выполнить операцию на данном устройстве, но некоторые входные тензоры отсутствуют на этом устройстве.
ExecutionEnvironment.Types