ประชาชน EnqueueTPUEmbeddingRaggedTensorBatch ชั้นสุดท้าย
ลดความยุ่งยากในการย้ายโค้ดที่ใช้ tf.nn.embedding_lookup()
sample_splits[i], embedding_indices[i] และ aggregation_weights[i] สอดคล้องกับคุณลักษณะ ith table_ids[i] ระบุว่าตารางการฝังตัวใดที่จะค้นหาคุณลักษณะ ith
เมตริกซ์ที่ตำแหน่งที่สอดคล้องกันในรายการอินพุตสองรายการ ได้แก่ embedding_indices และ aggregation_weights ต้องมีรูปร่างเหมือนกัน นั่นคืออันดับ 1 ที่มี dim_size() เท่ากับจำนวนการค้นหาทั้งหมดในตารางที่อธิบายโดยคุณลักษณะที่เกี่ยวข้อง
คลาสที่ซ้อนกัน
| ระดับ | EnqueueTPUEการฝังRaggedTensorBatch.Options | คุณลักษณะที่จำเป็นสำหรับการ EnqueueTPUEmbeddingRaggedTensorBatch | |
วิธีการสาธารณะ
| คง EnqueueTPUEmbeddingRaggedTensorBatch.Options | combiners (รายการ <String> combiners) |
| คง <T ขยายจำนวนยูขยายจำนวน, V ขยายจำนวน> EnqueueTPUEmbeddingRaggedTensorBatch | สร้าง ( ขอบเขต ขอบเขต Iterable < Operand <T >> sampleSplits, Iterable < Operand <U >> embeddingIndices, Iterable < Operand <V >> aggregationWeights, Operand <String> modeOverride รายการ <ยาว> tableIds, ตัวเลือก ... ตัวเลือก) วิธีการจากโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ EnqueueTPUEmbeddingRaggedTensorBatch ใหม่ |
| คง EnqueueTPUEmbeddingRaggedTensorBatch.Options | deviceOrdinal (Long deviceOrdinal) |
| คง EnqueueTPUEmbeddingRaggedTensorBatch.Options | maxSequenceLengths (รายการ <ยาว> maxSequenceLengths) |
| คง EnqueueTPUEmbeddingRaggedTensorBatch.Options | numFeatures (รายการ <ยาว> numFeatures) |
วิธีการสืบทอด
วิธีการสาธารณะ
สาธารณะคง EnqueueTPUEmbeddingRaggedTensorBatch.Options combiners (รายการ <String> combiners)
พารามิเตอร์
| เครื่องผสม | รายการของสเกลาร์สตริง หนึ่งรายการสำหรับแต่ละตารางการฝังที่ระบุวิธีการทำให้การเปิดใช้งานการฝังเป็นปกติหลังจากการรวมแบบถ่วงน้ำหนัก ตัวรวมที่รองรับคือ 'mean', 'sum' หรือ 'sqrtn' ไม่ถูกต้องที่จะมีผลรวมของน้ำหนักเป็น 0 สำหรับ 'ค่าเฉลี่ย' หรือผลรวมของน้ำหนักยกกำลังสองเป็น 0 สำหรับ 'sqrtn' หากไม่ผ่านตัวรวม ค่าเริ่มต้นคือการใช้ 'ผลรวม' สำหรับทุกตาราง |
|---|
สาธารณะคง EnqueueTPUEmbeddingRaggedTensorBatch สร้าง ( ขอบเขต ขอบเขต Iterable < Operand <T >> sampleSplits, Iterable < Operand <U >> embeddingIndices, Iterable < Operand <V >> aggregationWeights, Operand <String> modeOverride รายการ <ยาว> tableIds, ตัวเลือก .. . ตัวเลือก)
วิธีการจากโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ EnqueueTPUEmbeddingRaggedTensorBatch ใหม่
พารามิเตอร์
| ขอบเขต | ขอบเขตปัจจุบัน |
|---|---|
| แยกตัวอย่าง | รายการเทนเซอร์อันดับ 1 ที่ระบุจุดพักสำหรับการแยก embedding_indices และ aggregation_weights เป็นแถว มันสอดคล้องกับ ids.row_splits ใน embedding_lookup() เมื่อ ids เป็น RaggedTensor |
| การฝังดัชนี | รายการเทนเซอร์อันดับ 1 ดัชนีลงในตารางการฝัง มันสอดคล้องกับ ids.values ใน embedding_lookup() เมื่อ ids เป็น RaggedTensor |
| การรวมน้ำหนัก | รายการเทนเซอร์อันดับ 1 ที่มีตัวอย่างน้ำหนักรวมต่อการฝึก มันสอดคล้องกับฟิลด์ค่าของ RaggedTensor ที่มี row_splits เดียวกันกับรหัสใน embedding_lookup() เมื่อ ids เป็น RaggedTensor |
| modeOverride | อินพุตสตริงที่แทนที่โหมดที่ระบุใน TPUEmbeddingConfiguration ค่าที่รองรับคือ {'unspecified', 'inference', 'training', 'backward_pass_only'} เมื่อตั้งค่าเป็น 'ไม่ระบุ' โหมดที่ตั้งค่าใน TPUEmbeddingConfiguration จะถูกใช้ มิฉะนั้น mode_override จะถูกใช้ |
| ตารางรหัส | รายการจำนวนเต็มที่ระบุตัวระบุของตารางการฝัง (ออฟเซ็ตของ TableDescriptor ใน TPUEmbeddingConfiguration) เพื่อค้นหาอินพุตที่เกี่ยวข้อง ค้นหาอินพุต ith โดยใช้ table_ids[i] ขนาดของรายการ table_ids ต้องเท่ากับ sample_indices, embedding_indices และ aggregation_weights |
| ตัวเลือก | ดำเนินการค่าแอตทริบิวต์ทางเลือก |
คืนสินค้า
- อินสแตนซ์ใหม่ของ EnqueueTPUEmbeddingRaggedTensorBatch
สาธารณะคง EnqueueTPUEmbeddingRaggedTensorBatch.Options deviceOrdinal (Long deviceOrdinal)
พารามิเตอร์
| อุปกรณ์ลำดับ | อุปกรณ์ TPU ที่จะใช้ ควรเป็น >= 0 และน้อยกว่าจำนวนคอร์ TPU ในงานที่วางโหนด |
|---|