org.tensorflow

Define classes para criar, salvar, carregar e executar modelos do TensorFlow.

AVISO: O API é atualmente experimental e não é coberto por TensorFlow garantias de estabilidade API . Veja README.md para instruções de instalação.

O LabelImage exemplo demonstra o uso desta API para imagens Classificar usando um pré-treinados Inception arquitetura de rede neural convolucional. Isso demonstra:

  • Construção de gráfico: usando a classe OperationBuilder para construir um gráfico para decodificar, redimensionar e normalizar uma imagem JPEG.
  • Carregamento de modelo: usando Graph.importGraphDef () para carregar um modelo de iniciação pré-treinado.
  • Execução do gráfico: usando uma sessão para executar os gráficos e encontrar o melhor rótulo para uma imagem.

Exemplos adicionais podem ser encontrados no tensorflow / modelos repositório GitHub.

Interfaces

ExecutionEnvironment Define um ambiente para a criação e execução de TensorFlow Operation s.
Graph.WhileSubgraphBuilder Usado para instanciar uma classe abstrata que sobrescreve o método buildSubgraph para construir um subgráfico condicional ou de corpo para um loop while.
Operando <T estende TType > Interface implementada por operandos de uma operação do TensorFlow.
Operação Executa computação em tensores.
OperationBuilder Um construtor de Operation s.
Tensor Uma matriz multidimensional com tipagem estática.

Aulas

ConcreteFunction Um gráfico que pode ser chamado como uma única função, com uma assinatura de entrada e saída.
DeviceSpec Representa uma especificação (possivelmente parcial) para um dispositivo TensorFlow.
DeviceSpec.Builder Uma classe Builder para a construção de DeviceSpec classe.
EagerSession Um ambiente para executar operações do TensorFlow com entusiasmo.
EagerSession.Options
Gráfico Um gráfico de fluxo de dados que representa uma computação do TensorFlow.
GraphOperation Implementação de uma Operation adicionado como um nó de um Graph .
GraphOperationBuilder Um OperationBuilder para adicionar GraphOperation s para um Graph .
Output <T estende TType > Um identificador simbólica de um tensor produzido por uma Operation .
RawTensor Um tensor cuja memória não foi mapeada para um espaço de dados diretamente acessível a partir da JVM.
SavedModelBundle SavedModelBundle representa um modelo carregado do armazenamento.
SavedModelBundle.Exporter Opções para exportar um SavedModel.
SavedModelBundle.Loader Opções para carregar um SavedModel.
Servidor Um servidor TensorFlow em processo, para uso em treinamento distribuído.
Sessão Driver para Graph execução.
Session.Run Tensores de saída e metadados obtidos ao executar uma sessão.
Session.Runner Executar Operation s e avaliar Tensors .
Assinatura Descrever as entradas e saídas de uma entidade executável, tais como um ConcreteFunction , entre outros metadados útil.
Signature.Builder Constrói uma nova assinatura de função.
Signature.TensorDescription
TensorFlow Métodos de utilitário estáticos que descrevem o tempo de execução do TensorFlow.
TensorMapper <T estende TType > Mapas da memória nativa de uma RawTensor a um espaço de dados digitado n-dimensional acessível a partir da JVM.

Enums

DeviceSpec.DeviceType
EagerSession.DevicePlacementPolicy Controla como agir quando tentamos executar uma operação em um determinado dispositivo, mas alguns tensores de entrada não estão nesse dispositivo.
ExecutionEnvironment.Types