ConcreteFunction

Invoca uma função com uma única entrada e saída.

O chamador é responsável por fechar todos os Tensores.

Invoca uma função.

O chamador é responsável por fechar todos os Tensores.

Crie uma função a partir de uma assinatura e uma sessão gráfica válida.

A função não será proprietária da sessão nem de seu gráfico, o que significa que seu tempo de vida pode se estender além do escopo da função. Portanto a função não precisa ser fechada após seu uso. Por exemplo:

try (Graph g = new Graph()) {
   Placeholder<TFloat32> input = tf.placeholder(TFloat32.class);
   Add<TFloat32> output = tf.math.add(input, tf.constant(2.0f));
   Signature signature = Signature.builder().input("x", input).output("y", output).build();

   try (Session s = new Session(g)) {
     // Auto-closing the function just as an example but this is not required since it has
     // no effect
     try (ConcreteFunction f = ConcreteFunction.create(signature, s);
         TFloat32 t = TFloat32.scalarOf(2.0f)) {
       assertEquals(4.0f, ((TFloat32)function.call(x)).getFloat());
     
     // Session s is still valid at this point
   }
   // Graph g is still valid at this point
 }
 }

Cria uma função construindo um novo gráfico.

O functionBuilder deve inicializar o gráfico da função a partir da instância ERROR(/Ops) fornecida e retornar uma assinatura válida que será usada para alimentar os tensores de entrada e buscar os tensores de saída na execução.

A função será a proprietária do novo gráfico e da sessão resultante. Portanto, a função deve ser incluída corretamente com um bloco try-with-resources para garantir que todos os recursos nativos serão liberados quando a função for descartada. Por exemplo:

public class MyModel {

   public static Signature addTwo(Ops tf) {
     Placeholder<TFloat32> input = tf.placeholder(TFloat32.class);
     Add<TFloat32> output = tf.math.add(input, tf.constant(2.0f));
     return Signature.builder("addTwo").input("x", input).output("y", output).build();
   

   public static void main(String args[]) {
     try (ConcreteFunction function = ConcreteFunction.create(MyModel::addTwo);
         TFloat32 x = TFloat32.scalarOf(2.0f)) {
       assertEquals(4.0f, ((TFloat32)function.call(x)).getFloat());
     }
   }
 }
 }

Crie uma função a partir de uma assinatura e de um gráfico existente.

A função manterá a propriedade da sessão usada para executar o gráfico, mas não do gráfico em si, o que significa que o tempo de vida deste último pode se estender além do escopo da função. Por exemplo:

try (Graph g = new Graph()) {
   Placeholder<TFloat32> input = tf.placeholder(TFloat32.class);
   Add<TFloat32> output = tf.math.add(input, tf.constant(2.0f));
   Signature signature = Signature.builder().input("x", input).output("y", output).build();

   try (ConcreteFunction f = ConcreteFunction.create(signature, g);
       TFloat32 x = TFloat32.scalarOf(2.0f)) {
     assertEquals(4.0f, ((TFloat32)function.call(x)).getFloat());
   
   // Graph g is still valid at this point
 }
 }

Retorna o gráfico desta função

Exporte esta função como um modelo salvo.

Este método é um atalho conveniente equivalente a SavedModel.exporter(exportDir).withFunction(this).export()

Retorna a sessão usada para executar o gráfico ao chamar esta função

Em geral, um usuário não precisa lidar diretamente com a sessão de uma função e, em vez disso, contar com call(Map) para executar o gráfico. Mas em alguns casos, o acesso direto à sessão pode ser necessário, pois permite mais opções de execução.

Retorna a assinatura desta função