คลาสสุดท้ายสาธารณะ EnqueueTPUmbeddingRaggedTensorBatch
ทำให้การย้ายโค้ดที่ใช้ tf.nn.embedding_lookup() ง่ายขึ้น
example_splits[i], embedding_indices[i] และ aggregation_weights[i] สอดคล้องกับคุณลักษณะ i table_ids[i] ระบุว่าตารางฝังตัวใดที่จะค้นหาคุณลักษณะนี้
เทนเซอร์ที่ตำแหน่งที่สอดคล้องกันในรายการอินพุตสองรายการ ได้แก่ embedding_indices และ aggregation_weights จะต้องมีรูปร่างเหมือนกัน กล่าวคือ อันดับ 1 โดยมี dim_size() เท่ากับจำนวนการค้นหาทั้งหมดในตารางที่อธิบายโดยคุณลักษณะที่เกี่ยวข้อง
คลาสที่ซ้อนกัน
| ระดับ | เข้าคิว TPUmbeddingRaggedTensorBatch.Options | แอตทริบิวต์ทางเลือกสำหรับ EnqueueTPUEmbeddingRaggedTensorBatch | |
ค่าคงที่
| สตริง | OP_NAME | ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow |
วิธีการสาธารณะ
| EnqueueTPUmbeddingRaggedTensorBatch.Options แบบคงที่ | ตัวรวม (ตัวรวมรายการ<String>) |
| คงที่ <T ขยาย TNumber , U ขยาย TNumber , V ขยาย TNumber > EnqueueTPUEmbeddingRaggedTensorBatch | สร้าง ( ขอบเขต ขอบเขต Iterable< Operand <T>> SampleSplits, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand < TString > modeOverride, List<Long> tableIds, Options... options) วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ EnqueueTPUmbeddingRaggedTensorBatch ใหม่ |
| EnqueueTPUmbeddingRaggedTensorBatch.Options แบบคงที่ | deviceOrdinal (อุปกรณ์ยาวOrdinal) |
| EnqueueTPUmbeddingRaggedTensorBatch.Options แบบคงที่ | maxSequenceLengths (รายการ <ยาว> maxSequenceLengths) |
วิธีการสืบทอด
ค่าคงที่
สตริงสุดท้ายแบบคงที่สาธารณะ OP_NAME
ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow
ค่าคงที่: "EnqueueTPUmbeddingRaggedTensorBatch"
วิธีการสาธารณะ
EnqueueTPUEmbeddingRaggedTensorBatch.Options แบบคงที่สาธารณะ (ตัวรวมรายการ <String> )
พารามิเตอร์
| เครื่องผสม | รายการสเกลาร์สตริง หนึ่งรายการสำหรับแต่ละตารางการฝังที่ระบุวิธีทำให้การเปิดใช้งานการฝังเป็นมาตรฐานหลังจากการรวมแบบถ่วงน้ำหนัก ตัวรวมที่รองรับคือ 'mean', 'sum' หรือ 'sqrtn' ไม่ถูกต้องที่จะให้ผลรวมของน้ำหนักเป็น 0 สำหรับ 'ค่าเฉลี่ย' หรือผลรวมของน้ำหนักกำลังสองเป็น 0 สำหรับ 'sqrtn' หากไม่ผ่านตัวรวม ค่าเริ่มต้นคือการใช้ 'ผลรวม' สำหรับตารางทั้งหมด |
|---|
EnqueueTPUEmbeddingRaggedTensorBatch แบบคงที่สาธารณะ สร้าง (ขอบเขต ขอบเขต Iterable < Operand <T>> SampleSplits, Iterable < Operand <U>> embeddingIndices, Iterable < Operand <V>> aggregationWeights, Operand < TString > modeOverride, List <Long> tableIds, ตัวเลือก .. . ตัวเลือก)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ EnqueueTPUmbeddingRaggedTensorBatch ใหม่
พารามิเตอร์
| ขอบเขต | ขอบเขตปัจจุบัน |
|---|---|
| ตัวอย่างการแยก | รายการเทนเซอร์อันดับ 1 ที่ระบุจุดพักสำหรับการแยก embedding_indices และ aggregation_weights ออกเป็นแถว มันสอดคล้องกับ ids.row_splits ใน embedding_lookup() เมื่อ ids เป็น RaggedTensor |
| การฝังIndices | รายชื่อเทนเซอร์อันดับ 1 ซึ่งจัดทำดัชนีไว้ในตารางที่ฝัง มันสอดคล้องกับ ids.values ใน embedding_lookup() เมื่อ ids เป็น RaggedTensor |
| น้ำหนักรวม | รายชื่อเทนเซอร์อันดับ 1 ที่มีน้ำหนักรวมต่อตัวอย่างการฝึกอบรม มันสอดคล้องกับฟิลด์ค่าของ RaggedTensor ที่มี row_splits เดียวกันกับ id ใน embedding_lookup() เมื่อ ids เป็น RaggedTensor |
| โหมดแทนที่ | อินพุตสตริงที่แทนที่โหมดที่ระบุใน TPUEmbeddingConfiguration ค่าที่รองรับคือ {'unspecified', 'inference', 'training', 'backward_pass_only'} เมื่อตั้งค่าเป็น "ไม่ระบุ" ระบบจะใช้โหมดที่ตั้งค่าใน TPUEmbeddingConfiguration มิฉะนั้นจะใช้ mode_override |
| รหัสตาราง | รายการจำนวนเต็มที่ระบุตัวระบุของตารางที่ฝัง (ออฟเซ็ตของ TableDescriptor ใน TPUEmbeddingConfiguration) เพื่อค้นหาอินพุตที่เกี่ยวข้อง อินพุต I ถูกค้นหาโดยใช้ table_ids[i] ขนาดของรายการ table_ids ต้องเท่ากับขนาด example_indices, embedding_indices และ aggregation_weights |
| ตัวเลือก | มีค่าแอตทริบิวต์ทางเลือก |
การส่งคืน
- อินสแตนซ์ใหม่ของ EnqueueTPUmbeddingRaggedTensorBatch
สาธารณะ EnqueueTPUmbeddingRaggedTensorBatch.Options อุปกรณ์Ordinal (อุปกรณ์ยาวOrdinal)
พารามิเตอร์
| อุปกรณ์ลำดับ | อุปกรณ์ TPU ที่จะใช้ ควรเป็น >= 0 และน้อยกว่าจำนวนแกน TPU ในงานที่วางโหนด |
|---|