ConcreteFunction

Вызывает функцию с одним входом и выходом.

Вызывающая сторона отвечает за закрытие всех тензоров.

Вызывает функцию.

Вызывающая сторона отвечает за закрытие всех тензоров.

Создайте функцию на основе подписи и действительного сеанса графа.

Функция не будет владеть ни сеансом, ни его графиком, а это означает, что их время жизни может выходить за рамки функции. Поэтому функцию не нужно закрывать после ее использования. Например:

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
 }
 }

Создает функцию путем построения нового графика.

functionBuilder должен инициализировать граф функции из предоставленного экземпляра ERROR(/Ops) и вернуть действительную подпись, которая будет использоваться для подачи входных тензоров и получения выходных тензоров при выполнении.

Функция будет владельцем нового графика и его результирующего сеанса. Следовательно, функция должна быть правильно заключена в блок try-with-resources, чтобы гарантировать, что все собственные ресурсы будут освобождены после отмены функции. Например:

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());
     }
   }
 }
 }

Создайте функцию на основе сигнатуры и существующего графика.

Функция сохранит право собственности на сеанс, используемый для запуска графа, но не на сам граф, а это означает, что время жизни последнего может выходить за рамки функции. Например:

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
 }
 }

Возвращает график этой функции

Экспортируйте эту функцию как сохраненную модель.

Этот метод является удобным ярлыком, эквивалентным SavedModel.exporter(exportDir).withFunction(this).export()

Возвращает сеанс, используемый для выполнения графика при вызове этой функции.

В общем, пользователю не нужно напрямую обрабатывать сеанс функции и вместо этого полагаться на call(Map) для выполнения графа. Но в некоторых случаях может потребоваться прямой доступ к сеансу, поскольку он предоставляет больше возможностей запуска.

Возвращает подпись этой функции