tensorflow::ops::RestoreV2

#include <io_ops.h>

กู้คืนเทนเซอร์จากจุดตรวจสอบ V2

สรุป

เพื่อให้สามารถใช้งานร่วมกับรูปแบบ V1 ได้ ปัจจุบัน Op นี้จึงอนุญาตให้กู้คืนจากจุดตรวจสอบ V1 ได้เช่นกัน:

  • การดำเนินการนี้จะพยายามค้นหาไฟล์ดัชนี V2 ที่ชี้โดย "prefix" ก่อน และหากพบแล้วจะดำเนินการอ่านไฟล์นั้นเป็นจุดตรวจสอบ V2
  • มิเช่นนั้น ระบบจะเรียกใช้เส้นทางการอ่าน V1 การพึ่งพาพฤติกรรมนี้ไม่เป็นที่แนะนำ เนื่องจากความสามารถในการย้อนกลับไปอ่าน V1 อาจถูกยกเลิกและลบออกในที่สุด

โดยค่าเริ่มต้น ฟังก์ชันนี้จะกู้คืนเทนเซอร์ที่ระบุชื่อทั้งหมด หากผู้เรียกต้องการกู้คืนส่วนเฉพาะของเทนเซอร์ที่จัดเก็บไว้ "shape_and_slices" จะต้องเป็นสตริงที่ไม่ว่างเปล่าและมีรูปแบบที่ถูกต้อง

ผู้เรียกใช้งานต้องตรวจสอบให้แน่ใจว่าเทนเซอร์ที่ระบุชื่อทั้งหมดได้รับการจัดเก็บไว้ในจุดตรวจสอบแล้ว

ข้อโต้แย้ง:

  • ขอบเขต: วัตถุ ขอบเขต
  • คำนำหน้า: ต้องมีองค์ประกอบเพียงหนึ่งเดียว คำนำหน้าของจุดตรวจสอบ V2
  • tensor_names: รูปร่าง {N} ชื่อของเทนเซอร์ที่จะถูกกู้คืน
  • shape_and_slices: รูปร่าง {N} ข้อมูลจำเพาะของส่วนย่อยของเทนเซอร์ที่จะถูกกู้คืน สตริง ว่าง แสดงว่าเทนเซอร์เหล่านั้นไม่ได้ถูกแบ่งส่วน
  • dtypes: shape {N} รายการ dtype ที่คาดหวังสำหรับเทนเซอร์ ต้องตรงกับที่จัดเก็บไว้ใน checkpoint

การส่งคืนสินค้า:

  • OutputList : รูปร่าง {N} เทนเซอร์ที่ได้รับการกู้คืน ซึ่งรูปร่างจะถูกอ่านจากจุดตรวจสอบโดยตรง

ตัวสร้างและตัวทำลาย

RestoreV2 (const :: tensorflow::Scope & scope, :: tensorflow::Input prefix, :: tensorflow::Input tensor_names, :: tensorflow::Input shape_and_slices, const DataTypeSlice & dtypes)

คุณลักษณะสาธารณะ

operation
tensors

หน้าที่สาธารณะ

operator[] (size_t index) const

คุณลักษณะสาธารณะ

การดำเนินการ

Operation operation

เทนเซอร์

::tensorflow::OutputList tensors

หน้าที่สาธารณะ

กู้คืน V2

 RestoreV2(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input prefix,
  ::tensorflow::Input tensor_names,
  ::tensorflow::Input shape_and_slices,
  const DataTypeSlice & dtypes
)

ผู้ดำเนินการ[]

::tensorflow::Output operator[](
  size_t index
) const