คลาสสุดท้ายสาธารณะ EnqueueTPUEmbeddingBatch
op ที่จัดคิวรายการของแบทช์เทนเซอร์อินพุตไปยัง TPUEmbedding
op ที่จัดคิวรายการของแบทช์เทนเซอร์อินพุตไปยัง TPUEmbedding
คลาสที่ซ้อนกัน
ระดับ | EnqueueTPUEmbeddingBatch.Options | แอตทริบิวต์เสริมสำหรับ EnqueueTPUEmbeddingBatch |
วิธีการสาธารณะ
คงที่ EnqueueTPUEmbeddingBatch.Options | ตัวรวม (รายการ <String> ตัวรวม) |
EnqueueTPUEmbeddingBatch แบบคงที่ | สร้าง ( ขอบเขต ขอบเขต Iterable< Operaand <String>> แบทช์ Operand <String> modeOverride ตัวเลือก... ) วิธีการจากโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ EnqueueTPUEmbeddingBatch ใหม่ |
คงที่ EnqueueTPUEmbeddingBatch.Options | deviceOrdinal (อุปกรณ์แบบยาวOrdinal) |
วิธีการสืบทอด
วิธีการสาธารณะ
EnqueueTPUEmbeddingBatch.Options แบบคงที่สาธารณะ (List <String> combiners )
พารามิเตอร์
เครื่องผสม | รายการของสเกลาร์สตริง หนึ่งรายการสำหรับแต่ละตารางการฝังที่ระบุวิธีการทำให้การเปิดใช้งานการฝังเป็นปกติหลังจากการรวมแบบถ่วงน้ำหนัก ตัวรวมที่รองรับคือ 'mean', 'sum' หรือ 'sqrtn' ไม่ถูกต้องที่จะมีผลรวมของน้ำหนักเป็น 0 สำหรับ 'ค่าเฉลี่ย' หรือผลรวมของน้ำหนักยกกำลังสองเป็น 0 สำหรับ 'sqrtn' หากไม่ผ่านตัวรวม ค่าเริ่มต้นคือการใช้ 'ผลรวม' สำหรับทุกตาราง |
---|
EnqueueTPUEmbeddingBatch สาธารณะแบบคงที่ สร้าง ( ขอบเขต ขอบเขต Iterable< Operand <String>> แบทช์ Operand <String> modeOverride ตัวเลือก... )
วิธีการจากโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ EnqueueTPUEmbeddingBatch ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|---|
แบทช์ | รายการเทนเซอร์ 1 มิติ หนึ่งรายการสำหรับแต่ละตารางการฝัง ซึ่งประกอบด้วยอินพุตแบทช์ที่เข้ารหัสเป็น dist_belief.SparseFeatures protos หากฟิลด์น้ำหนักในโปรโต SparseFeatures ไม่ได้เติมข้อมูลสำหรับ ID ระบบจะถือว่าน้ำหนักเป็น 1.0 |
modeOverride | อินพุตสตริงที่แทนที่โหมดที่ระบุใน TPUEmbeddingConfiguration ค่าที่รองรับคือ {'unspecified', 'inference', 'training', 'backward_pass_only'} เมื่อตั้งค่าเป็น 'ไม่ระบุ' โหมดที่ตั้งค่าใน TPUEmbeddingConfiguration จะถูกใช้ มิฉะนั้น mode_override จะถูกใช้ |
ตัวเลือก | ดำเนินการค่าแอตทริบิวต์ทางเลือก |
คืนสินค้า
- อินสแตนซ์ใหม่ของ EnqueueTPUEmbeddingBatch
EnqueueTPUEmbeddingBatch.Options deviceOrdinal แบบ คงที่สาธารณะ (Long deviceOrdinal)
พารามิเตอร์
อุปกรณ์ลำดับ | อุปกรณ์ TPU ที่จะใช้ ซึ่งควรเป็น -1 เมื่อ Op ทำงานบนอุปกรณ์ TPU และ >= 0 เมื่อ Op ทำงานบนอุปกรณ์ CPU |
---|