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