ปฏิบัติการที่ดึงพารามิเตอร์การปรับให้เหมาะสมจากการฝังไปยังหน่วยความจำโฮสต์
ปฏิบัติการที่ดึงพารามิเตอร์การปรับให้เหมาะสมจากการฝังไปยังหน่วยความจำโฮสต์ ต้องนำหน้าด้วยตัวเลือก ConfigureTPUEmbeddingHost ที่ตั้งค่าการกำหนดค่าตารางฝังที่ถูกต้อง ตัวอย่างเช่น op นี้ใช้เพื่อดึงพารามิเตอร์ที่อัปเดตก่อนที่จะบันทึกจุดตรวจสอบ สำหรับ Adagrad นั้น auxiliary1 จะมีตัวสะสมหลังจากรัน op นี้ สำหรับ SGD ค่าเสริม* ทั้งหมดจะว่างเปล่า (0x0 เทนเซอร์สำหรับตารางนั้น) สำหรับ FTRL นั้น auxiliary1 จะมีตัวสะสม และ auxiliary2 จะมีเงื่อนไขเชิงเส้น สำหรับ ADAM ส่วนเสริม 1 จะมีโมเมนต้า ส่วนส่วนเสริม 2 จะมีความเร็ว
วิธีการสาธารณะ
รายการ < เอาต์พุต <ลอย>> | เสริม1 () รายการเทนเซอร์ หนึ่งรายการสำหรับแต่ละตารางที่ฝัง ซึ่งมีพารามิเตอร์การปรับให้เหมาะสมเสริมตัวแรกที่เก็บไว้ |
รายการ < เอาต์พุต <ลอย>> | เสริม2 () รายการเทนเซอร์ หนึ่งรายการสำหรับแต่ละตารางที่ฝัง ซึ่งมีพารามิเตอร์การปรับให้เหมาะสมเสริมที่สองเก็บไว้ |
รายการ < เอาต์พุต <ลอย>> | เสริม3 () รายการเทนเซอร์ หนึ่งรายการสำหรับแต่ละตารางที่ฝัง ซึ่งมีพารามิเตอร์การปรับให้เหมาะสมเสริมตัวที่สามเก็บไว้ |
รายการ < เอาต์พุต <ลอย>> | เสริม4 () รายการเทนเซอร์ หนึ่งรายการสำหรับแต่ละตารางที่ฝัง ซึ่งมีพารามิเตอร์การปรับให้เหมาะสมเสริมที่สี่เก็บไว้ |
รายการ < เอาต์พุต <ลอย>> | เสริม5 () รายการเทนเซอร์ หนึ่งรายการสำหรับแต่ละตารางที่ฝัง ซึ่งมีพารามิเตอร์การปรับให้เหมาะสมเสริมที่ห้าเก็บไว้ |
รายการ < เอาต์พุต <ลอย>> | เสริม6 () รายการเทนเซอร์ หนึ่งรายการสำหรับแต่ละตารางที่ฝัง ซึ่งมีพารามิเตอร์การปรับให้เหมาะสมเสริมหกรายการเก็บไว้ |
รายการ < เอาต์พุต <ลอย>> | เสริม7 () รายการเทนเซอร์ หนึ่งรายการสำหรับแต่ละตารางที่ฝัง ซึ่งมีพารามิเตอร์การปรับให้เหมาะสมเสริมที่เจ็ดเก็บไว้ |
คงพารามิเตอร์ การดึงข้อมูล AllTPUEmbeddingParameters | |
รายการ < เอาต์พุต <ลอย>> | พารามิเตอร์ () รายการเทนเซอร์ หนึ่งรายการสำหรับแต่ละตารางการฝัง ซึ่งมีพารามิเตอร์ตารางการฝังที่เก็บไว้ |
วิธีการสืบทอด
วิธีการสาธารณะ
รายการสาธารณะ < เอาต์พุต <Float>> auxiliary1 ()
รายการเทนเซอร์ หนึ่งรายการสำหรับแต่ละตารางที่ฝัง ซึ่งมีพารามิเตอร์การปรับให้เหมาะสมเสริมตัวแรกที่เก็บไว้ องค์ประกอบมีอยู่ในรายการ แต่มีขนาดเป็นศูนย์สำหรับพารามิเตอร์การปรับให้เหมาะสมที่ไม่ได้ใช้ (ขึ้นอยู่กับอัลกอริทึมที่ใช้สำหรับแต่ละตาราง)
รายการสาธารณะ < เอาต์พุต <Float>> auxiliary2 ()
รายการเทนเซอร์ หนึ่งรายการสำหรับแต่ละตารางที่ฝัง ซึ่งมีพารามิเตอร์การปรับให้เหมาะสมเสริมที่สองเก็บไว้ องค์ประกอบมีอยู่ในรายการ แต่มีขนาดเป็นศูนย์สำหรับพารามิเตอร์การปรับให้เหมาะสมที่ไม่ได้ใช้ (ขึ้นอยู่กับอัลกอริทึมที่ใช้สำหรับแต่ละตาราง)
รายการสาธารณะ < เอาต์พุต <ลอย>> auxiliary3 ()
รายการเทนเซอร์ หนึ่งรายการสำหรับแต่ละตารางที่ฝัง ซึ่งมีพารามิเตอร์การปรับให้เหมาะสมเสริมตัวที่สามเก็บไว้ องค์ประกอบมีอยู่ในรายการ แต่มีขนาดเป็นศูนย์สำหรับพารามิเตอร์การปรับให้เหมาะสมที่ไม่ได้ใช้ (ขึ้นอยู่กับอัลกอริทึมที่ใช้สำหรับแต่ละตาราง)
รายการสาธารณะ < เอาต์พุต <Float>> auxiliary4 ()
รายการเทนเซอร์ หนึ่งรายการสำหรับแต่ละตารางที่ฝัง ซึ่งมีพารามิเตอร์การปรับให้เหมาะสมเสริมที่สี่เก็บไว้ องค์ประกอบมีอยู่ในรายการ แต่มีขนาดเป็นศูนย์สำหรับพารามิเตอร์การปรับให้เหมาะสมที่ไม่ได้ใช้ (ขึ้นอยู่กับอัลกอริทึมที่ใช้สำหรับแต่ละตาราง)
รายการสาธารณะ < เอาท์พุต <ลอย>> ช่วย 5 ()
รายการเทนเซอร์ หนึ่งรายการสำหรับแต่ละตารางที่ฝัง ซึ่งมีพารามิเตอร์การปรับให้เหมาะสมเสริมที่ห้าเก็บไว้ องค์ประกอบมีอยู่ในรายการ แต่มีขนาดเป็นศูนย์สำหรับพารามิเตอร์การปรับให้เหมาะสมที่ไม่ได้ใช้ (ขึ้นอยู่กับอัลกอริทึมที่ใช้สำหรับแต่ละตาราง)
รายการสาธารณะ < เอาต์พุต <Float>> auxiliary6 ()
รายการเทนเซอร์ หนึ่งรายการสำหรับแต่ละตารางที่ฝัง ซึ่งมีพารามิเตอร์การปรับให้เหมาะสมเสริมหกรายการเก็บไว้ องค์ประกอบมีอยู่ในรายการ แต่มีขนาดเป็นศูนย์สำหรับพารามิเตอร์การปรับให้เหมาะสมที่ไม่ได้ใช้ (ขึ้นอยู่กับอัลกอริทึมที่ใช้สำหรับแต่ละตาราง)
รายการสาธารณะ < เอาต์พุต <Float>> auxiliary7 ()
รายการเทนเซอร์ หนึ่งรายการสำหรับแต่ละตารางที่ฝัง ซึ่งมีพารามิเตอร์การปรับให้เหมาะสมเสริมที่เจ็ดเก็บไว้ องค์ประกอบมีอยู่ในรายการ แต่มีขนาดเป็นศูนย์สำหรับพารามิเตอร์การปรับให้เหมาะสมที่ไม่ได้ใช้ (ขึ้นอยู่กับอัลกอริทึมที่ใช้สำหรับแต่ละตาราง)
สร้าง พารามิเตอร์การดึงข้อมูล แบบคงที่สาธารณะแบบคงที่ (ขอบเขต ขอบเขต , ตาราง NumTable แบบยาว, การกำหนดค่าสตริง, แบบยาว numShards, แบบยาว shardId)
วิธีการจากโรงงานเพื่อสร้างคลาสที่ห่อการดำเนินการ RestoreeAllTPUEmbeddingParameters ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|---|
ตารางตัวเลข | จำนวนตารางที่ฝัง |
กำหนดค่า | โปรโต TPUEmbeddingConfiguration อธิบายพารามิเตอร์ตารางที่กำลังโหลด และซีเรียลไลซ์เป็นสตริง |
numShards | จำนวนชาร์ดที่ใช้แบ่งตารางที่ฝังไว้ |
shardId | ตัวระบุชาร์ดสำหรับการดำเนินการนี้ |
การส่งคืน
- อินสแตนซ์ใหม่ของ RestoreeAllTPUEmbeddingParameters
รายการสาธารณะ < เอาท์พุท <ลอย>> พารามิเตอร์ ()
รายการเทนเซอร์ หนึ่งรายการสำหรับแต่ละตารางการฝัง ซึ่งมีพารามิเตอร์ตารางการฝังที่เก็บไว้