เทนเซอร์โฟลว์:: ปฏิบัติการ:: คืนค่า

#include <io_ops.h>

คืนค่าเทนเซอร์จากไฟล์จุดตรวจ

สรุป

อ่านเทนเซอร์ที่จัดเก็บไว้ในไฟล์เดียวหรือหลายไฟล์ หากมีไฟล์หลายไฟล์ (เช่น เนื่องจากเมตริกซ์ถูกบันทึกเป็นชิ้น) file_pattern อาจมีสัญลักษณ์ตัวแทน ( * และ ? ) ในส่วนชื่อไฟล์เท่านั้น ไม่ใช่ในส่วนไดเร็กทอรี

หาก file_pattern ตรงกับหลายไฟล์ ก็สามารถใช้เพื่อบอก preferred_shard นัยว่าไฟล์ใดที่มีแนวโน้มว่าจะพบเทนเซอร์ที่ร้องขอ การดำเนินการนี้จะเปิดไฟล์ที่ดัชนี preferred_shard ในรายการไฟล์ที่ตรงกันก่อน และพยายามกู้คืนเทนเซอร์จากไฟล์นั้น เฉพาะในกรณีที่ไม่พบเทนเซอร์หรือเทนเซอร์บางส่วนในไฟล์แรกนั้น Op จะเปิดไฟล์ทั้งหมด การตั้ง preferred_shard ให้ตรงกับค่าที่ส่งผ่านเนื่องจากอินพุต shard ของ Save Op ที่ตรงกันอาจทำให้ Restore เร็วขึ้น คุณลักษณะนี้ส่งผลต่อประสิทธิภาพเท่านั้น ไม่ใช่ความถูกต้อง ค่าเริ่มต้น -1 หมายถึงไฟล์ได้รับการประมวลผลตามลำดับ

ดูเพิ่มเติม RestoreSlice

ข้อโต้แย้ง:

  • ขอบเขต: วัตถุ ขอบเขต
  • file_pattern: ต้องมีองค์ประกอบเดียว รูปแบบของไฟล์ที่เราอ่านเทนเซอร์
  • tensor_name: ต้องมีองค์ประกอบเดียว ชื่อของเทนเซอร์ที่จะกู้คืน
  • dt: ประเภทของเทนเซอร์ที่จะคืนค่า

แอ็ตทริบิวต์ทางเลือก (ดู Attrs ):

  • ที่ต้องการ_shard: ดัชนีของไฟล์ที่จะเปิดก่อนหากหลายไฟล์ตรงกับ file_pattern

ผลตอบแทน:

  • Output : เทนเซอร์ที่ได้รับการฟื้นฟู

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

Restore (const :: tensorflow::Scope & scope, :: tensorflow::Input file_pattern, :: tensorflow::Input tensor_name, DataType dt)
Restore (const :: tensorflow::Scope & scope, :: tensorflow::Input file_pattern, :: tensorflow::Input tensor_name, DataType dt, const Restore::Attrs & attrs)

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

operation
tensor

งานสาธารณะ

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

ฟังก์ชันคงที่สาธารณะ

PreferredShard (int64 x)

โครงสร้าง

เทนเซอร์โฟลว์ :: ops :: คืนค่า :: Attrs

ตัวตั้งค่าแอ็ตทริบิวต์ทางเลือกสำหรับ Restore

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

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

Operation operation

เทนเซอร์

::tensorflow::Output tensor

งานสาธารณะ

คืนค่า

 Restore(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input file_pattern,
  ::tensorflow::Input tensor_name,
  DataType dt
)

คืนค่า

 Restore(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input file_pattern,
  ::tensorflow::Input tensor_name,
  DataType dt,
  const Restore::Attrs & attrs
)

โหนด

::tensorflow::Node * node() const 

ตัวดำเนินการ::tensorflow::อินพุต

 operator::tensorflow::Input() const 

ตัวดำเนินการ::tensorflow::เอาต์พุต

 operator::tensorflow::Output() const 

ฟังก์ชันคงที่สาธารณะ

PreferredShard

Attrs PreferredShard(
  int64 x
)