सार्वजनिक अंतिम वर्ग EnqueueTPUEmbeddingRaggedTensorBatch
tf.nn.embedding_lookup() का उपयोग करने वाले कोड की पोर्टिंग को आसान बनाता है।
नमूना_स्प्लिट्स[i], एम्बेडिंग_इंडिसेस[i] और एग्रीगेशन_वेट्स[i] ith फीचर के अनुरूप हैं। table_ids[i] इंगित करता है कि ith सुविधा को देखने के लिए कौन सी एम्बेडिंग तालिका है।
दो इनपुट सूचियों, एम्बेडिंग_इंडिस और एग्रीगेशन_वेट्स में संबंधित पदों पर टेंसर का आकार समान होना चाहिए, यानी रैंक 1 के साथ डिम_साइज () संबंधित फीचर द्वारा वर्णित तालिका में लुकअप की कुल संख्या के बराबर है।
नेस्टेड कक्षाएं
| कक्षा | EnqueueTPUEएम्बेडिंगRaggedTensorBatch.Options | के लिए वैकल्पिक विशेषताओं EnqueueTPUEmbeddingRaggedTensorBatch | |
सार्वजनिक तरीके
| स्थिर EnqueueTPUEmbeddingRaggedTensorBatch.Options | समेलक (सूची <स्ट्रिंग> समेलक) |
| स्थिर <टी फैली संख्या, यू संख्या फैली हुई है, वी फैली संख्या> EnqueueTPUEmbeddingRaggedTensorBatch | बनाने ( स्कोप गुंजाइश, Iterable < ओपेरैंड <टी >> sampleSplits, Iterable < ओपेरैंड <यू >> embeddingIndices, Iterable < ओपेरैंड <वी >> aggregationWeights, ओपेरैंड <string> modeOverride, सूची <लांग> tableIds, विकल्प ... विकल्प) एक नया EnqueueTPUEmbeddingRaggedTensorBatch ऑपरेशन रैपिंग क्लास बनाने के लिए फ़ैक्टरी विधि। |
| स्थिर EnqueueTPUEmbeddingRaggedTensorBatch.Options | deviceOrdinal (लांग deviceOrdinal) |
| स्थिर EnqueueTPUEmbeddingRaggedTensorBatch.Options | maxSequenceLengths (सूची <लांग> maxSequenceLengths) |
| स्थिर EnqueueTPUEmbeddingRaggedTensorBatch.Options | numFeatures (सूची <लांग> numFeatures) |
विरासत में मिली विधियां
सार्वजनिक तरीके
सार्वजनिक स्थिर EnqueueTPUEmbeddingRaggedTensorBatch.Options समेलक (सूची <स्ट्रिंग> समेलक)
मापदंडों
| समेलक | स्ट्रिंग स्केलर्स की एक सूची, प्रत्येक एम्बेडिंग तालिका के लिए एक जो निर्दिष्ट करती है कि भारित योग के बाद एम्बेडिंग सक्रियण को कैसे सामान्य किया जाए। समर्थित संयोजक 'माध्य', 'योग' या 'वर्ग' हैं। 'माध्य' के लिए भारों का योग 0 होना या वर्ग भार का योग 'sqrtn' के लिए 0 होना अमान्य है। यदि संयोजक पारित नहीं होते हैं, तो डिफ़ॉल्ट सभी तालिकाओं के लिए 'योग' का उपयोग करना है। |
|---|
सार्वजनिक स्थिर EnqueueTPUEmbeddingRaggedTensorBatch (बनाने स्कोप गुंजाइश, Iterable < ओपेरैंड <टी >> sampleSplits, Iterable < ओपेरैंड <यू >> embeddingIndices, Iterable < ओपेरैंड <वी >> aggregationWeights, ओपेरैंड <string> modeOverride, सूची <लांग> tableIds, विकल्प .. । विकल्प)
एक नया EnqueueTPUEmbeddingRaggedTensorBatch ऑपरेशन रैपिंग क्लास बनाने के लिए फ़ैक्टरी विधि।
मापदंडों
| दायरा | वर्तमान दायरा |
|---|---|
| नमूना विभाजन | एम्बेडिंग_इंडिस और एग्रीगेशन_वेट को पंक्तियों में विभाजित करने के लिए ब्रेक पॉइंट निर्दिष्ट करने वाले रैंक 1 टेंसर की एक सूची। यह एम्बेडिंग_लुकअप () में ids.row_splits से मेल खाती है, जब आईडी एक रैग्ड टेंसर होता है। |
| एम्बेडिंग इंडेक्स | रैंक 1 टेंसर की एक सूची, एम्बेडिंग टेबल में इंडेक्स। यह एम्बेडिंग_लुकअप () में ids.values से मेल खाता है, जब आईडी एक रैग्ड टेंसर होता है। |
| एकत्रीकरणवजन | प्रति प्रशिक्षण उदाहरण एकत्रीकरण भार वाले रैंक 1 टेंसर की एक सूची। यह एक रैग्डटेन्सर के मान फ़ील्ड से मेल खाता है जिसमें समान पंक्ति_स्प्लिट्स एम्बेडिंग_लुकअप () में आईडी के रूप में होते हैं, जब आईडी एक रैग्ड टेंसर होता है। |
| मोड ओवरराइड | एक स्ट्रिंग इनपुट जो TPUEmbeddingConfiguration में निर्दिष्ट मोड को ओवरराइड करता है। समर्थित मान {'अनिर्दिष्ट', 'अनुमान', 'प्रशिक्षण', 'backward_pass_only'} हैं। जब 'अनिर्दिष्ट' पर सेट किया जाता है, तो TPUEmbeddingConfiguration में सेट मोड का उपयोग किया जाता है, अन्यथा मोड_ओवरराइड का उपयोग किया जाता है। |
| तालिका आईडी | एम्बेडिंग तालिका के पहचानकर्ता को निर्दिष्ट करने वाले पूर्णांकों की एक सूची (TPUEmbeddingConfiguration में TableDescriptor की ऑफसेट) संबंधित इनपुट को देखने के लिए। ith इनपुट को table_ids[i] का उपयोग करके देखा जाता है। टेबल_आईड्स सूची का आकार नमूना_इंडिस, एम्बेडिंग_इंडिस और एग्रीगेशन_वेट के बराबर होना चाहिए। |
| विकल्प | वैकल्पिक विशेषता मान रखता है |
रिटर्न
- EnqueueTPUEएम्बेडिंगRaggedTensorBatch का एक नया उदाहरण
सार्वजनिक स्थिर EnqueueTPUEmbeddingRaggedTensorBatch.Options deviceOrdinal (लांग deviceOrdinal)
मापदंडों
| डिवाइसऑर्डिनल | उपयोग करने के लिए टीपीयू डिवाइस। जिस कार्य पर नोड रखा गया है, उसमें टीपीयू कोर की संख्या> = 0 और उससे कम होनी चाहिए। |
|---|