EnqueueTPUEmbeddingIntegerBatch

คลาสสุดท้ายสาธารณะ 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 ในงานที่วางโหนด