การดำเนินการที่ทำให้เกิดความแตกต่างของ TPU Embeddings
op นี้เพียงส่งคืนอินพุตแรก ซึ่งถือว่าถูกตัดออกจากเทนเซอร์ที่ส่งคืนโดย TPUEmbeddingDequeueActivations การมีอยู่ของ op นี้และอาร์กิวเมนต์แรกของมันคือตัวแปรที่ฝึกได้ ช่วยให้สามารถสร้างความแตกต่างอัตโนมัติของกราฟที่มีการฝังผ่านไลบรารี TPU Embedding Python
วิธีการสาธารณะ
เอาท์พุต <โฟลต> | เป็นเอาท์พุต () ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์ |
การเปิดใช้งาน TPUEmbedding แบบคงที่ | |
เอาท์พุต <โฟลต> | เอาท์พุท () |
วิธีการสืบทอด
วิธีการสาธารณะ
เอาท์ พุท สาธารณะ <Float> asOutput ()
ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์
อินพุตสำหรับการดำเนินการ TensorFlow คือเอาต์พุตของการดำเนินการ TensorFlow อื่น วิธีการนี้ใช้เพื่อรับหมายเลขอ้างอิงสัญลักษณ์ที่แสดงถึงการคำนวณอินพุต
การสร้าง TPUEmbeddingActivations แบบคงที่สาธารณะ (ขอบเขต ขอบเขต , ตัวดำเนินการ <Float> embeddingVariable, ตัวดำเนินการ <Float> การเปิดใช้งานแบบแบ่งส่วน, Long tableId, Long lookupId)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ TPUEmbeddingActivations ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|---|
การฝังตัวแปร | ตัวแปรที่สามารถฝึกได้ ช่วยให้เครื่องมือเพิ่มประสิทธิภาพสามารถค้นหาการดำเนินการนี้ได้ |
การเปิดใช้งานหั่นบาง ๆ | การเปิดใช้งานเทนเซอร์แบบฝังที่จะกลับมา |
รหัสตาราง | รหัสของตารางในการกำหนดค่าเลเยอร์การฝังซึ่งมีการคำนวณการเปิดใช้งานเหล่านี้ |
รหัสการค้นหา | ตัวระบุชุดของดัชนีการฝังซึ่งสร้างการเปิดใช้งานเหล่านี้ |
การส่งคืน
- อินสแตนซ์ใหม่ของ TPUEmbeddingActivations