เทนเซอร์โฟลว์:: ปฏิบัติการ:: คืนค่า
#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 )