सार्वजनिक अंतिम वर्ग EnqueueTPUEएम्बेडिंगस्पार्सबैच
एक ऑप जो एक SparseTensor से TPUE एम्बेडिंग इनपुट इंडेक्स को एनक्यू करता है।
यह ऑप कोड की पोर्टिंग को आसान बनाता है जो एम्बेडिंग_लुकअप_स्पार्स () का उपयोग करता है, हालांकि एम्बेडिंग_लुकअप_स्पार्स () के लिए स्पार्सटेन्सर तर्कों के कुछ पायथन प्रीप्रोसेसिंग को इस ओप के लिए तर्क उत्पन्न करने की आवश्यकता होती है, क्योंकि प्रति प्रशिक्षण चरण में केवल एक एनक्यूयूटीपीयूएम्बेडिंगस्पार्सबैच ओप की अनुमति है।
तीन इनपुट सूचियों में संबंधित पदों पर टेंसर का आकार समान होना चाहिए, यानी रैंक 1 के साथ dim_size() संबंधित तालिका_आईडी द्वारा वर्णित तालिका में लुकअप की कुल संख्या के बराबर।
नेस्टेड कक्षाएं
| कक्षा | EnqueueTPUEएम्बेडिंगSparseBatch.Options | EnqueueTPUEmbeddingSparseBatch के लिए वैकल्पिक विशेषताएं | |
सार्वजनिक तरीके
| स्थैतिक EnqueueTPUEएम्बेडिंगSparseBatch.Options | संयोजक (सूची<स्ट्रिंग> संयोजक) |
| स्थिर <T संख्या बढ़ाता है, U संख्या बढ़ाता है, V संख्या बढ़ाता है> EnqueueTPUEmbeddingSparseBatch | |
| स्थैतिक EnqueueTPUEएम्बेडिंगSparseBatch.Options | डिवाइसऑर्डिनल (लंबी डिवाइसऑर्डिनल) |
विरासत में मिली विधियां
सार्वजनिक तरीके
सार्वजनिक स्थैतिक EnqueueTPUEएम्बेडिंगस्पार्सबैच.विकल्प संयोजक (सूची<स्ट्रिंग> संयोजक)
मापदंडों
| समेलक | स्ट्रिंग स्केलर्स की एक सूची, प्रत्येक एम्बेडिंग तालिका के लिए एक जो निर्दिष्ट करती है कि भारित योग के बाद एम्बेडिंग सक्रियण को कैसे सामान्य किया जाए। समर्थित संयोजक 'माध्य', 'योग' या 'वर्ग' हैं। 'माध्य' के लिए भारों का योग 0 होना या वर्ग भार का योग 'sqrtn' के लिए 0 होना अमान्य है। यदि संयोजक पारित नहीं होते हैं, तो डिफ़ॉल्ट सभी तालिकाओं के लिए 'योग' का उपयोग करना है। |
|---|
सार्वजनिक स्थैतिक EnqueueTPUEएंबेडिंगस्पार्सबैच क्रिएट ( स्कोप स्कोप, इटरेबल< ऑपरेंड <टी>> सैंपलइंडिस, इटरेबल< ऑपरेंड <यू>> एम्बेडिंगइंडिस, इटरेबल < ऑपरेंड <वी>> एग्रीगेशनवेट्स, ऑपरेंड <स्ट्रिंग> मोड ओवरराइड, विकल्प ... विकल्प)
एक नया EnqueueTPUEmbeddingSparseBatch ऑपरेशन रैपिंग क्लास बनाने के लिए फ़ैक्टरी विधि।
मापदंडों
| दायरा | वर्तमान दायरा |
|---|---|
| नमूना सूचकांक | प्रशिक्षण उदाहरण और विशेषता को निर्दिष्ट करने वाले रैंक 1 टेंसर की एक सूची जिससे संबंधित एम्बेडिंग_इंडिस और एग्रीगेशन_वेट मान संबंधित हैं। नमूना_इंडिस [i] को बी * एनएफ + एफ के बराबर होना चाहिए, जहां एनएफ संबंधित तालिका से सुविधाओं की संख्या है, एफ [0, एनएफ) में है, और बी [0, बैच आकार में है)। |
| एम्बेडिंग इंडेक्स | रैंक 1 टेंसर की एक सूची, एम्बेडिंग टेबल में इंडेक्स। |
| एकत्रीकरणवजन | प्रति नमूना युक्त रैंक 1 टेंसर की एक सूची - यानी प्रति (प्रशिक्षण उदाहरण, सुविधा) - एकत्रीकरण भार। |
| मोड ओवरराइड | एक स्ट्रिंग इनपुट जो TPUEmbeddingConfiguration में निर्दिष्ट मोड को ओवरराइड करता है। समर्थित मान {'अनिर्दिष्ट', 'अनुमान', 'प्रशिक्षण', 'backward_pass_only'} हैं। जब 'अनिर्दिष्ट' पर सेट किया जाता है, तो TPUEmbeddingConfiguration में सेट मोड का उपयोग किया जाता है, अन्यथा मोड_ओवरराइड का उपयोग किया जाता है। |
| विकल्प | वैकल्पिक विशेषता मान रखता है |
रिटर्न
- EnqueueTPUEएम्बेडिंगस्पार्सबैच का एक नया उदाहरण
सार्वजनिक स्थैतिक EnqueueTPUE एम्बेडिंगस्पार्सबैच.विकल्प डिवाइसऑर्डिनल (लंबी डिवाइसऑर्डिनल)
मापदंडों
| डिवाइसऑर्डिनल | उपयोग करने के लिए टीपीयू डिवाइस। जिस कार्य पर नोड रखा गया है, उसमें टीपीयू कोर की संख्या> = 0 और उससे कम होनी चाहिए। |
|---|