EnqueueTPUEmbeddingSparseBatch

คลาสสุดท้ายสาธารณะ EnqueueTPUmbeddingSparseBatch

การดำเนินการที่จัดคิวดัชนีอินพุต TPUEmbedding จาก SparseTensor

Op นี้ทำให้การย้ายโค้ดที่ใช้ embedding_lookup_sparse() ง่ายขึ้น แม้ว่าการประมวลผล Python ล่วงหน้าของอาร์กิวเมนต์ SparseTensor เป็น embedding_lookup_sparse() จะต้องสร้างอาร์กิวเมนต์ให้กับ Op นี้ เนื่องจากอนุญาตให้ใช้ EnqueueTPUEmbeddingSparseBatch Op เดียวเท่านั้นต่อขั้นตอนการฝึก

เทนเซอร์ที่ตำแหน่งที่สอดคล้องกันในรายการอินพุตทั้งสามรายการจะต้องมีรูปร่างเหมือนกัน กล่าวคือ อันดับ 1 โดยมี dim_size() เท่ากับจำนวนการค้นหาทั้งหมดในตารางที่อธิบายโดย table_id ที่สอดคล้องกัน

คลาสที่ซ้อนกัน

ระดับ จัดคิวTPUmbeddingSparseBatch.Options แอตทริบิวต์เพิ่มเติมสำหรับ EnqueueTPUEmbeddingSparseBatch

วิธีการสาธารณะ

EnqueueTPUmbeddingSparseBatch.Options แบบคงที่
ตัวรวม (ตัวรวมรายการ<String>)
คงที่ <T ขยายหมายเลข, U ขยายหมายเลข, V ขยายหมายเลข> EnqueueTPUmbeddingSparseBatch
สร้าง ( ขอบเขตขอบเขต Iterable< Operand <T>> SampleIndices, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand <String> modeOverride, Options... options)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ EnqueueTPUEmbeddingSparseBatch ใหม่
EnqueueTPUmbeddingSparseBatch.Options แบบคงที่
deviceOrdinal (อุปกรณ์ยาวOrdinal)

วิธีการสืบทอด

วิธีการสาธารณะ

ตัวรวม EnqueueTPUEmbeddingSparseBatch.Options สาธารณะแบบคงที่ (ตัวรวมรายการ <สตริง>)

พารามิเตอร์
เครื่องผสม รายการสเกลาร์สตริง หนึ่งรายการสำหรับแต่ละตารางการฝังที่ระบุวิธีทำให้การเปิดใช้งานการฝังเป็นมาตรฐานหลังจากการรวมแบบถ่วงน้ำหนัก ตัวรวมที่รองรับคือ 'mean', 'sum' หรือ 'sqrtn' ไม่ถูกต้องที่จะให้ผลรวมของน้ำหนักเป็น 0 สำหรับ 'ค่าเฉลี่ย' หรือผลรวมของน้ำหนักกำลังสองเป็น 0 สำหรับ 'sqrtn' หากไม่ผ่านตัวรวม ค่าเริ่มต้นคือการใช้ 'ผลรวม' สำหรับตารางทั้งหมด

สร้าง EnqueueTPUEmbeddingSparseBatch แบบคงที่สาธารณะ (ขอบเขต ขอบเขต , Iterable< Operand <T>> SampleIndices, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand <String> modeOverride, ตัวเลือก... ตัวเลือก)

วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ EnqueueTPUEmbeddingSparseBatch ใหม่

พารามิเตอร์
ขอบเขต ขอบเขตปัจจุบัน
ตัวอย่างดัชนี รายการเทนเซอร์อันดับ 1 ที่ระบุตัวอย่างการฝึกอบรมและฟีเจอร์ซึ่งมีค่า embedding_indices และ aggregation_weights ที่สอดคล้องกัน Sample_indices[i] ต้องเท่ากับ b * nf + f โดยที่ nf คือจำนวนคุณลักษณะจากตารางที่เกี่ยวข้อง f อยู่ใน [0, nf) และ b อยู่ใน [0 ขนาดแบทช์)
การฝังIndices รายชื่อเทนเซอร์อันดับ 1 ซึ่งจัดทำดัชนีไว้ในตารางที่ฝัง
น้ำหนักรวม รายการเทนเซอร์อันดับ 1 ที่มีตุ้มน้ำหนักรวมต่อตัวอย่าง เช่น ต่อ (ตัวอย่างการฝึกอบรม คุณลักษณะ)
โหมดแทนที่ อินพุตสตริงที่แทนที่โหมดที่ระบุใน TPUEmbeddingConfiguration ค่าที่รองรับคือ {'unspecified', 'inference', 'training', 'backward_pass_only'} เมื่อตั้งค่าเป็น "ไม่ระบุ" ระบบจะใช้โหมดที่ตั้งค่าใน TPUEmbeddingConfiguration มิฉะนั้นจะใช้ mode_override
ตัวเลือก มีค่าแอตทริบิวต์ทางเลือก
การส่งคืน
  • อินสแตนซ์ใหม่ของ EnqueueTPUmbeddingSparseBatch

EnqueueTPUEmbeddingSparseBatch.Options สาธารณะแบบคง ที่อุปกรณ์Ordinal (อุปกรณ์แบบยาวOrdinal)

พารามิเตอร์
อุปกรณ์ลำดับ อุปกรณ์ TPU ที่จะใช้ ควรเป็น >= 0 และน้อยกว่าจำนวนแกน TPU ในงานที่วางโหนด