EnqueueTPUEmbeddingSparseTensorBatch

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

ทำให้การย้ายโค้ดที่ใช้ tf.nn.embedding_lookup_sparse() ง่ายขึ้น

example_indices[i], embedding_indices[i] และ aggregation_weights[i] สอดคล้องกับคุณลักษณะ i table_ids[i] ระบุว่าตารางฝังตัวใดที่จะค้นหาคุณลักษณะนี้

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

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

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

ค่าคงที่

สตริง OP_NAME ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow

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

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

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

ค่าคงที่

สตริงสุดท้ายแบบคงที่สาธารณะ OP_NAME

ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow

ค่าคงที่: "EnqueueTPUmbeddingSparseTensorBatch"

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

สาธารณะ EnqueueTPUEmbeddingSparseTensorBatch.Options ตัวรวม (ตัวรวมรายการ <สตริง>)

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

สาธารณะ EnqueueTPUEmbeddingSparseTensorBatch สร้าง (ขอบเขต ขอบเขต Iterable < Operand <T>> SampleIndices, Iterable < Operand <U>> embeddingIndices, Iterable < Operand <V>> aggregationWeights, Operand < TString > modeOverride, List <Long> tableIds, ตัวเลือก .. . ตัวเลือก)

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

พารามิเตอร์
ขอบเขต ขอบเขตปัจจุบัน
ตัวอย่างดัชนี รายการเทนเซอร์อันดับ 1 ที่ระบุตัวอย่างการฝึกอบรมซึ่งมีค่า embedding_indices และ aggregation_weights ที่สอดคล้องกัน โดยสอดคล้องกับ sp_ids.indices[:,0] ใน embedding_lookup_sparse()
การฝังIndices รายชื่อเทนเซอร์อันดับ 1 ซึ่งจัดทำดัชนีไว้ในตารางที่ฝัง ซึ่งสอดคล้องกับ sp_ids.values ​​ใน embedding_lookup_sparse()
น้ำหนักรวม รายชื่อเทนเซอร์อันดับ 1 ที่มีน้ำหนักรวมต่อตัวอย่างการฝึกอบรม ซึ่งสอดคล้องกับ sp_weights.values ​​ใน embedding_lookup_sparse()
โหมดแทนที่ อินพุตสตริงที่แทนที่โหมดที่ระบุใน TPUEmbeddingConfiguration ค่าที่รองรับคือ {'unspecified', 'inference', 'training', 'backward_pass_only'} เมื่อตั้งค่าเป็น "ไม่ระบุ" ระบบจะใช้โหมดที่ตั้งค่าใน TPUEmbeddingConfiguration มิฉะนั้นจะใช้ mode_override
รหัสตาราง รายการจำนวนเต็มที่ระบุตัวระบุของตารางที่ฝัง (ออฟเซ็ตของ TableDescriptor ใน TPUEmbeddingConfiguration) เพื่อค้นหาอินพุตที่เกี่ยวข้อง ค้นหาอินพุตที่ i โดยใช้ table_ids[i] ขนาดของรายการ table_ids ต้องเท่ากับขนาด example_indices, embedding_indices และ aggregation_weights
ตัวเลือก มีค่าแอตทริบิวต์ทางเลือก
การส่งคืน
  • อินสแตนซ์ใหม่ของ EnqueueTPUmbeddingSparseTensorBatch

สาธารณะ EnqueueTPUEmbeddingSparseTensorBatch.Options อุปกรณ์Ordinal (อุปกรณ์ยาวOrdinal)

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

สาธารณะ EnqueueTPUEmbeddingSparseTensorBatch.Options maxSequenceLengths (รายการ <ยาว> maxSequenceLengths)