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