ConcreteFunction

Memanggil fungsi dengan satu input dan output.

Penelepon bertanggung jawab untuk menutup semua Tensor.

Memanggil suatu fungsi.

Penelepon bertanggung jawab untuk menutup semua Tensor.

Buat fungsi dari tanda tangan dan sesi grafik yang valid.

Fungsi tersebut tidak akan memiliki sesi maupun grafiknya, yang berarti masa pakainya dapat melampaui cakupan fungsi tersebut. Oleh karena itu fungsinya tidak perlu ditutup setelah penggunaannya. Misalnya:

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

Membuat fungsi dengan membuat grafik baru.

functionBuilder harus menginisialisasi grafik fungsi dari instance ERROR(/Ops) yang disediakan dan mengembalikan tanda tangan valid yang akan digunakan untuk memasukkan tensor masukan dan mengambil tensor keluaran saat eksekusi.

Fungsi tersebut akan menjadi pemilik grafik baru dan sesi yang dihasilkannya. Oleh karena itu, fungsi tersebut harus diapit dengan benar dengan blok try-with-resources untuk menjamin bahwa semua sumber daya asli akan dibebaskan setelah fungsi tersebut dibuang. Misalnya:

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

Buat fungsi dari tanda tangan dan grafik yang ada.

Fungsi tersebut akan mempertahankan kepemilikan sesi yang digunakan untuk menjalankan grafik, tetapi tidak mempertahankan kepemilikan grafik itu sendiri, yang berarti bahwa masa pakai grafik tersebut dapat melampaui cakupan fungsi tersebut. Misalnya:

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

Mengembalikan grafik fungsi ini

Ekspor fungsi ini sebagai model tersimpan.

Metode ini merupakan pintasan praktis yang setara dengan SavedModel.exporter(exportDir).withFunction(this).export()

Mengembalikan sesi yang digunakan untuk mengeksekusi grafik saat memanggil fungsi ini

Secara umum, pengguna tidak perlu menangani sesi suatu fungsi secara langsung dan mengandalkan call(Map) untuk mengeksekusi grafik. Namun dalam beberapa kasus, akses langsung ke sesi mungkin diperlukan, karena memungkinkan lebih banyak opsi berjalan.

Mengembalikan tanda tangan dari fungsi ini