คลาสสุดท้ายสาธารณะ EnqueueTPUmbeddingIntegerBatch
การดำเนินการที่จัดคิวรายการเทนเซอร์แบทช์อินพุตเป็น TPUEmbedding
คลาสที่ซ้อนกัน
ระดับ | เข้าคิว TPUmbeddingIntegerBatch.Options | แอตทริบิวต์เพิ่มเติมสำหรับ EnqueueTPUEmbeddingIntegerBatch |
วิธีการสาธารณะ
EnqueueTPUembeddingIntegerBatch แบบคงที่ | สร้าง (ขอบเขต ขอบเขต Iterable< Operand <Integer>> แบตช์ Operand <String> modeOverride ตัวเลือก... ตัวเลือก) วิธีการจากโรงงานเพื่อสร้างคลาสที่ห่อการดำเนินการ EnqueueTPUEmbeddingIntegerBatch ใหม่ |
EnqueueTPUmbeddingIntegerBatch.Options แบบคงที่ | deviceOrdinal (อุปกรณ์ยาวOrdinal) |
วิธีการสืบทอด
วิธีการสาธารณะ
สร้าง EnqueueTPUEmbeddingIntegerBatch แบบคงที่สาธารณะ (ขอบเขต ขอบเขต , Iterable < ตัวดำเนินการ <Integer>> ชุด, ตัวดำเนินการ <String> modeOverride, ตัวเลือก... ตัวเลือก)
วิธีการจากโรงงานเพื่อสร้างคลาสที่ห่อการดำเนินการ EnqueueTPUEmbeddingIntegerBatch ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|---|
แบทช์ | รายการเทนเซอร์ 1D หนึ่งรายการสำหรับแต่ละตารางที่ฝัง ซึ่งมีดัชนีลงในตาราง |
โหมดแทนที่ | อินพุตสตริงที่แทนที่โหมดที่ระบุใน TPUEmbeddingConfiguration ค่าที่รองรับคือ {'unspecified', 'inference', 'training', 'backward_pass_only'} เมื่อตั้งค่าเป็น "ไม่ระบุ" ระบบจะใช้โหมดที่ตั้งค่าใน TPUEmbeddingConfiguration มิฉะนั้นจะใช้ mode_override |
ตัวเลือก | มีค่าแอตทริบิวต์ทางเลือก |
การส่งคืน
- อินสแตนซ์ใหม่ของ EnqueueTPUmbeddingIntegerBatch
สาธารณะ EnqueueTPUEmbeddingIntegerBatch.Options อุปกรณ์Ordinal (อุปกรณ์ยาวOrdinal)
พารามิเตอร์
อุปกรณ์ลำดับ | อุปกรณ์ TPU ที่จะใช้ ควรเป็น >= 0 และน้อยกว่าจำนวนแกน TPU ในงานที่วางโหนด |
---|