op ที่ได้รับการเปิดใช้งานการฝังบน TPU
ระบบ TPU ดำเนินการค้นหาการฝังและการรวมที่ระบุโดยอาร์กิวเมนต์ไปยัง TPUEmbeddingEnqueue(Integer/Sparse/SparseTensor)Batch ผลลัพธ์ของการรวมเหล่านี้สามารถมองเห็นได้ในกราฟ Tensorflow เป็นผลลัพธ์ของ RecvTPUEmbeddingActivations op op นี้ส่งคืนรายการที่มีหนึ่ง Tensor ของการเปิดใช้งานต่อตารางที่ระบุในโมเดล RecvTPUEmbeddingActivations op ในกราฟ TPU มีได้ไม่เกิน 1 รายการ
วิธีการสาธารณะ
RecvTPUEmbeddingActivations แบบคงที่ | |
Iterator < ตัวถูกดำเนินการ <Float>> | ตัววนซ้ำ () |
รายการ< เอาท์พุต <โฟลต>> | เอาท์พุท () TensorList ของการเปิดใช้การฝังที่มีหนึ่ง Tensor ต่อตารางการฝังในโมเดล |
วิธีการสืบทอด
วิธีการสาธารณะ
RecvTPUEmbeddingActivations สาธารณะแบบคงที่ สร้าง ( ขอบเขต ขอบเขต, numOutputs แบบยาว, การกำหนดค่าสตริง)
วิธีการจากโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ RecvTPUEmbeddingActivations ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|---|
numOutputs | จำนวนเทนเซอร์การเปิดใช้งานเอาต์พุต เท่ากับจำนวนตารางการฝังในโมเดล |
config | โปรโตคอนฟิกูเรชันการกำหนดค่า TPUEmbedding แบบอนุกรม |
คืนสินค้า
- อินสแตนซ์ใหม่ของ RecvTPUEmbeddingActivations
รายการสาธารณะ < เอาท์พุต <โฟลต>> เอาท์พุท ()
TensorList ของการเปิดใช้การฝังที่มีหนึ่ง Tensor ต่อตารางการฝังในโมเดล