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