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