ประชาชน EnqueueTPUEmbeddingSparseTensorBatch ชั้นสุดท้าย
ลดความยุ่งยากในการย้ายโค้ดที่ใช้ tf.nn.embedding_lookup_sparse()
sample_indices[i], embedding_indices[i] และ aggregation_weights[i] สอดคล้องกับคุณลักษณะ ith table_ids[i] ระบุว่าตารางการฝังตัวใดที่จะค้นหาคุณลักษณะ ith
เทนเซอร์ที่ตำแหน่งที่สอดคล้องกันในรายการอินพุตสามรายการ (sample_indices, embedding_indices และ aggregation_weights) ต้องมีรูปร่างเหมือนกัน นั่นคืออันดับ 1 ที่มี dim_size() เท่ากับจำนวนการค้นหาทั้งหมดในตารางที่อธิบายโดยฟีเจอร์ที่เกี่ยวข้อง
คลาสที่ซ้อนกัน
| ระดับ | EnqueueTPUEการฝังSparseTensorBatch.Options | คุณลักษณะที่จำเป็นสำหรับการ EnqueueTPUEmbeddingSparseTensorBatch | |
วิธีการสาธารณะ
| คง EnqueueTPUEmbeddingSparseTensorBatch.Options | combiners (รายการ <String> combiners) |
| คง <T ขยายจำนวนยูขยายจำนวน, V ขยายจำนวน> EnqueueTPUEmbeddingSparseTensorBatch | สร้าง ( ขอบเขต ขอบเขต Iterable < Operand <T >> sampleIndices, Iterable < Operand <U >> embeddingIndices, Iterable < Operand <V >> aggregationWeights, Operand <String> modeOverride รายการ <ยาว> tableIds, ตัวเลือก ... ตัวเลือก) วิธีการจากโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ EnqueueTPUEmbeddingSparseTensorBatch ใหม่ |
| คง EnqueueTPUEmbeddingSparseTensorBatch.Options | deviceOrdinal (Long deviceOrdinal) |
| คง EnqueueTPUEmbeddingSparseTensorBatch.Options | maxSequenceLengths (รายการ <ยาว> maxSequenceLengths) |
| คง EnqueueTPUEmbeddingSparseTensorBatch.Options | numFeatures (รายการ <ยาว> numFeatures) |
วิธีการสืบทอด
วิธีการสาธารณะ
สาธารณะคง EnqueueTPUEmbeddingSparseTensorBatch.Options combiners (รายการ <String> combiners)
พารามิเตอร์
| เครื่องผสม | รายการของสเกลาร์สตริง หนึ่งรายการสำหรับแต่ละตารางการฝังที่ระบุวิธีการทำให้การเปิดใช้งานการฝังเป็นปกติหลังจากการรวมแบบถ่วงน้ำหนัก ตัวรวมที่รองรับคือ 'mean', 'sum' หรือ 'sqrtn' ไม่ถูกต้องที่จะมีผลรวมของน้ำหนักเป็น 0 สำหรับ 'ค่าเฉลี่ย' หรือผลรวมของน้ำหนักยกกำลังสองเป็น 0 สำหรับ 'sqrtn' หากไม่ผ่านตัวรวม ค่าเริ่มต้นคือการใช้ 'ผลรวม' สำหรับทุกตาราง |
|---|
สาธารณะคง EnqueueTPUEmbeddingSparseTensorBatch สร้าง ( ขอบเขต ขอบเขต Iterable < Operand <T >> sampleIndices, Iterable < Operand <U >> embeddingIndices, Iterable < Operand <V >> aggregationWeights, Operand <String> modeOverride รายการ <ยาว> tableIds, ตัวเลือก .. . ตัวเลือก)
วิธีการจากโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ EnqueueTPUEmbeddingSparseTensorBatch ใหม่
พารามิเตอร์
| ขอบเขต | ขอบเขตปัจจุบัน |
|---|---|
| ดัชนีตัวอย่าง | รายการเทนเซอร์อันดับ 1 ที่ระบุตัวอย่างการฝึกที่มีค่า embedding_indices และ aggregation_weights ที่สอดคล้องกัน มันสอดคล้องกับ sp_ids.indices[:,0] ใน embedding_lookup_sparse() |
| การฝังดัชนี | รายการเทนเซอร์อันดับ 1 ดัชนีลงในตารางการฝัง มันสอดคล้องกับ sp_ids.values ใน embedding_lookup_sparse() |
| การรวมน้ำหนัก | รายการเทนเซอร์อันดับ 1 ที่มีตัวอย่างน้ำหนักรวมต่อการฝึก มันสอดคล้องกับ sp_weights.values ใน embedding_lookup_sparse() |
| 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 |
| ตัวเลือก | ดำเนินการค่าแอตทริบิวต์ทางเลือก |
คืนสินค้า
- อินสแตนซ์ใหม่ของ EnqueueTPUEmbeddingSparseTensorBatch
สาธารณะคง EnqueueTPUEmbeddingSparseTensorBatch.Options deviceOrdinal (Long deviceOrdinal)
พารามิเตอร์
| อุปกรณ์ลำดับ | อุปกรณ์ TPU ที่จะใช้ ควรเป็น >= 0 และน้อยกว่าจำนวนคอร์ TPU ในงานที่วางโหนด |
|---|