Init

ลงทะเบียน op เป็นตัวเริ่มต้นของกราฟ

ตัวเริ่มต้นที่ลงทะเบียนแล้วจะถูกจัดกลุ่มเป็นหน่วยการคำนวณเดียวโดยการเพิ่มและดำเนินการการดำเนินการ init จากเซสชันกราฟ นี่เป็นสิ่งที่ไม่ต้องดำเนินการหากดำเนินการในเซสชันที่กระตือรือร้น

วิธีการจากโรงงานเพื่อสร้างการดำเนินการที่เรียกใช้งานตัวเริ่มต้นทั้งหมดของกราฟ

ตัวเริ่มต้นทั้งหมดที่เพิ่มลงในกราฟผ่าน tf.initAdd จะถูกจัดกลุ่มเข้าด้วยกันเป็นหน่วยการคำนวณเดียวในกราฟ จากนั้นจะต้องเพิ่มการดำเนินการนี้ลงในกราฟใดๆ โดยใช้ variables ตั้งแต่หนึ่งตัวขึ้นไป และดำเนินการหนึ่งครั้งก่อนที่จะรันกราฟ เพื่อให้สถานะของตัวแปรเริ่มต้นได้อย่างเหมาะสม

เมื่อกราฟถูกสร้างขึ้นโดยกระบวนการเดียวกันกับที่กำลังรันเซสชัน ตัวเริ่มต้นสามารถเรียกใช้ได้โดยการดำเนินการจุดสิ้นสุดเดียวนี้ ตัวอย่างเช่น:

try (Graph g = new Graph()) {
   Variable<TInt32> x = tf.variable(tf.constant(10));  // initAdd is called implicitly
   Variable<TInt32> y = tf.variable(tf.constant(20));  // idem
   Add<TInt32> z = tf.math.add(x, y);

   try (Session s = new Session(g)) {
     s.run(tf.init());  // initialize all variables

     try (TInt32 t = (TInt32)s.runner().fetch(z).run().get(0)) {
       assertEquals(30, t.data().getInt());
     
   }
 }
 }

เมื่อกราฟถูกสร้างขึ้นโดยกระบวนการที่แยกจากกัน คุณสามารถเรียกใช้ตัวเริ่มต้นได้โดยการเรียกใช้ init op ตามชื่อ ซึ่งมีค่าเริ่มต้นเป็น DEFAULT_NAME ตัวอย่างเช่น:

// Building the model
 try (Graph g = new Graph()) {
   Variable<TInt32> x = tf.variable(tf.constant(10));  // initAdd is called implicitly
   Variable<TInt32> y = tf.variable(tf.constant(20));  // idem
   Add<TInt32> z = tf.withName("z").math.add(x, y);

   tf.init();  // add variables initializers to the graph, as Init.DEFAULT_NAME
   // ...exporting graph as a saved model...
 

 ...

 // Running the model
 try (SavedModelBundle model = SavedModelBundle.load("/path/to/model", "train")) {
   model.session().run(Init.DEFAULT_NAME);

   try (TInt32 t = (TInt32)s.runner().fetch("z").run().get(0)) {
     assertEquals(30, t.data().getInt());
   }
 }
 }