एक ऑप जो ऑप्टिमाइज़ेशन पैरामीटर को एम्बेडिंग से होस्ट मेमोरी में पुनर्प्राप्त करता है।
एक ऑप जो ऑप्टिमाइज़ेशन पैरामीटर को एम्बेडिंग से होस्ट मेमोरी में पुनर्प्राप्त करता है। एक ConfigureTPUEmbeddingHost op से पहले होना चाहिए जो सही एम्बेडिंग तालिका कॉन्फ़िगरेशन सेट करता है। उदाहरण के लिए, इस ऑप का उपयोग चेकपॉइंट को सहेजने से पहले अद्यतन मापदंडों को पुनः प्राप्त करने के लिए किया जाता है। Adagrad के लिए, इस ऑप को चलाने के बाद सहायक 1 में संचायक होंगे। SGD के लिए, सभी सहायक* मान खाली होंगे (उस तालिका के लिए 0x0 टेंसर)। FTRL के लिए, सहायक 1 में संचायक होंगे और सहायक 2 में रैखिक शब्द होंगे। ADAM के लिए, सहायक 1 में गति होगी और सहायक 2 में वेग होंगे।
सार्वजनिक तरीके
| सूची < आउटपुट <फ्लोट>> | सहायक 1 () टेनर्स की एक सूची, प्रत्येक एम्बेडिंग टेबल के लिए एक, जिसमें पहला सहायक ऑप्टिमाइज़ेशन पैरामीटर संग्रहीत होता है। |
| सूची < आउटपुट <फ्लोट>> | सहायक 2 () दसियों की एक सूची, प्रत्येक एम्बेडिंग तालिका के लिए एक, जिसमें दूसरा सहायक अनुकूलन पैरामीटर संग्रहीत होता है। |
| सूची < आउटपुट <फ्लोट>> | सहायक3 () टेन्सर्स की एक सूची, प्रत्येक एम्बेडिंग टेबल के लिए एक, जिसमें तीसरा सहायक ऑप्टिमाइज़ेशन पैरामीटर संग्रहीत होता है। |
| सूची < आउटपुट <फ्लोट>> | सहायक4 () टेनर्स की एक सूची, प्रत्येक एम्बेडिंग तालिका के लिए एक, जिसमें चौथा सहायक अनुकूलन पैरामीटर संग्रहीत होता है। |
| सूची < आउटपुट <फ्लोट>> | सहायक5 () टेंसर की एक सूची, प्रत्येक एम्बेडिंग तालिका के लिए एक, जिसमें पाँचवाँ सहायक अनुकूलन पैरामीटर संग्रहीत होता है। |
| सूची < आउटपुट <फ्लोट>> | सहायक6 () टेंसर की एक सूची, प्रत्येक एम्बेडिंग तालिका के लिए एक, जिसमें छह सहायक अनुकूलन पैरामीटर संग्रहीत होते हैं। |
| सूची < आउटपुट <फ्लोट>> | सहायक7 () टेंसर की एक सूची, प्रत्येक एम्बेडिंग तालिका के लिए एक, जिसमें सातवां सहायक अनुकूलन पैरामीटर संग्रहीत है। |
| स्थिर पुनर्प्राप्ति AllTPUEएम्बेडिंगपैरामीटर | |
| सूची < आउटपुट <फ्लोट>> | पैरामीटर () प्रत्येक एम्बेडिंग तालिका के लिए टेंसर की एक सूची, जिसमें संग्रहीत एम्बेडिंग तालिका पैरामीटर होते हैं। |
विरासत में मिली विधियां
सार्वजनिक तरीके
सार्वजनिक सूची < आउटपुट <फ्लोट>> सहायक 1 ()
टेनर्स की एक सूची, प्रत्येक एम्बेडिंग टेबल के लिए एक, जिसमें पहला सहायक ऑप्टिमाइज़ेशन पैरामीटर संग्रहीत होता है। तत्व सूची में मौजूद हैं, लेकिन अप्रयुक्त अनुकूलन पैरामीटर (प्रत्येक तालिका के लिए उपयोग में एल्गोरिदम के आधार पर) के लिए शून्य आकार है।
सार्वजनिक सूची < आउटपुट <फ्लोट>> सहायक 2 ()
दसियों की एक सूची, प्रत्येक एम्बेडिंग तालिका के लिए एक, जिसमें दूसरा सहायक अनुकूलन पैरामीटर संग्रहीत होता है। तत्व सूची में मौजूद हैं, लेकिन अप्रयुक्त अनुकूलन पैरामीटर (प्रत्येक तालिका के लिए उपयोग में एल्गोरिदम के आधार पर) के लिए शून्य आकार है।
सार्वजनिक सूची < आउटपुट <फ्लोट>> सहायक3 ()
टेन्सर्स की एक सूची, प्रत्येक एम्बेडिंग टेबल के लिए एक, जिसमें तीसरा सहायक ऑप्टिमाइज़ेशन पैरामीटर संग्रहीत होता है। तत्व सूची में मौजूद हैं, लेकिन अप्रयुक्त अनुकूलन पैरामीटर (प्रत्येक तालिका के लिए उपयोग में एल्गोरिदम के आधार पर) के लिए शून्य आकार है।
सार्वजनिक सूची < आउटपुट <फ्लोट>> सहायक 4 ()
टेनर्स की एक सूची, प्रत्येक एम्बेडिंग तालिका के लिए एक, जिसमें चौथा सहायक अनुकूलन पैरामीटर संग्रहीत होता है। तत्व सूची में मौजूद हैं, लेकिन अप्रयुक्त अनुकूलन पैरामीटर (प्रत्येक तालिका के लिए उपयोग में एल्गोरिदम के आधार पर) के लिए शून्य आकार है।
सार्वजनिक सूची < आउटपुट <फ्लोट>> सहायक 5 ()
टेंसर की एक सूची, प्रत्येक एम्बेडिंग तालिका के लिए एक, जिसमें पाँचवाँ सहायक अनुकूलन पैरामीटर संग्रहीत होता है। तत्व सूची में मौजूद हैं, लेकिन अप्रयुक्त अनुकूलन पैरामीटर (प्रत्येक तालिका के लिए उपयोग में एल्गोरिदम के आधार पर) के लिए शून्य आकार है।
सार्वजनिक सूची < आउटपुट <फ्लोट>> सहायक6 ()
टेंसर की एक सूची, प्रत्येक एम्बेडिंग तालिका के लिए एक, जिसमें छह सहायक अनुकूलन पैरामीटर संग्रहीत होते हैं। तत्व सूची में मौजूद हैं, लेकिन अप्रयुक्त अनुकूलन पैरामीटर (प्रत्येक तालिका के लिए उपयोग में एल्गोरिदम के आधार पर) के लिए शून्य आकार है।
सार्वजनिक सूची < आउटपुट <फ्लोट>> सहायक7 ()
टेंसर की एक सूची, प्रत्येक एम्बेडिंग तालिका के लिए एक, जिसमें सातवां सहायक अनुकूलन पैरामीटर संग्रहीत है। तत्व सूची में मौजूद हैं, लेकिन अप्रयुक्त अनुकूलन पैरामीटर (प्रत्येक तालिका के लिए उपयोग में एल्गोरिदम के आधार पर) के लिए शून्य आकार है।
पब्लिक स्टैटिक रिट्रीव ऑलटीपीयूएम्बेडिंगपैरामीटर्स क्रिएट ( स्कोप स्कोप, लॉन्ग न्यूमटेबल्स, स्ट्रिंग कॉन्फिगरेशन, लॉन्ग numShards, लॉन्ग shardId)
एक नया RetrieveAllTPUEmbeddingParameters ऑपरेशन रैपिंग क्लास बनाने के लिए फ़ैक्टरी विधि।
मापदंडों
| दायरा | वर्तमान दायरा |
|---|---|
| अंक तालिका | एम्बेडिंग टेबल की संख्या। |
| कॉन्फ़िग | एक TPUEmbeddingConfiguration प्रोटो जो लोड किए जा रहे तालिका मापदंडों का वर्णन करता है, एक स्ट्रिंग के लिए क्रमबद्ध। |
| numShards | शार्ड्स की संख्या जिसमें एम्बेडिंग टेबल विभाजित हैं। |
| शार्प आईडी | इस ऑपरेशन के लिए शार्ड की पहचान। |
रिटर्न
- RetrieveAllTPUEएम्बेडिंगपैरामीटर का एक नया उदाहरण
सार्वजनिक सूची < आउटपुट <फ्लोट>> पैरामीटर ()
प्रत्येक एम्बेडिंग तालिका के लिए टेंसर की एक सूची, जिसमें संग्रहीत एम्बेडिंग तालिका पैरामीटर होते हैं।