ConcreteFunction

Gọi một hàm có một đầu vào và đầu ra duy nhất.

Người gọi có trách nhiệm đóng tất cả Tensors.

Gọi một hàm.

Người gọi có trách nhiệm đóng tất cả Tensors.

Tạo một hàm từ chữ ký và phiên biểu đồ hợp lệ.

Hàm sẽ không sở hữu phiên cũng như biểu đồ của nó, nghĩa là thời gian tồn tại của chúng có thể vượt ra ngoài phạm vi của hàm. Do đó, chức năng này không cần phải đóng sau khi sử dụng. Ví dụ:

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

Tạo một hàm bằng cách xây dựng một biểu đồ mới.

functionBuilder phải khởi tạo biểu đồ hàm từ phiên bản ERROR(/Ops) được cung cấp và trả về một chữ ký hợp lệ sẽ được sử dụng để cung cấp các tensor đầu vào và tìm nạp các tensor đầu ra khi thực thi.

Hàm sẽ là chủ sở hữu của biểu đồ mới và phiên kết quả của nó. Do đó, hàm phải được bao bọc đúng cách bằng khối try-with-resources để đảm bảo rằng tất cả tài nguyên gốc sẽ được giải phóng sau khi hàm bị loại bỏ. Ví dụ:

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

Tạo một hàm từ chữ ký và biểu đồ hiện có.

Hàm sẽ giữ quyền sở hữu phiên được sử dụng để chạy biểu đồ chứ không phải bản thân biểu đồ, nghĩa là thời gian tồn tại của biểu đồ sau có thể vượt ra ngoài phạm vi của hàm. Ví dụ:

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

Trả về đồ thị của hàm này

Xuất chức năng này dưới dạng mô hình đã lưu.

Phương thức này là lối tắt thuận tiện tương đương với SavedModel.exporter(exportDir).withFunction(this).export()

Trả về phiên được sử dụng để thực thi biểu đồ khi gọi hàm này

Nói chung, người dùng không cần phải xử lý trực tiếp phiên của một hàm và thay vào đó dựa vào call(Map) để thực thi biểu đồ. Nhưng trong một số trường hợp, việc truy cập trực tiếp vào phiên có thể cần thiết vì nó cho phép nhiều tùy chọn chạy hơn.

Trả về chữ ký của hàm này