एक ऑप जो टीपीयू पर एम्बेडिंग सक्रियण प्राप्त करता है।
टीपीयू प्रणाली एम्बेडिंग लुकअप और एकत्रीकरण करती है। इन एकत्रीकरणों के परिणाम XlaRecvTPUEmbeddingActivations Op के आउटपुट के रूप में Tensorflow ग्राफ़ पर दिखाई देते हैं। यह ऑप मॉडल में निर्दिष्ट प्रति तालिका सक्रियणों के एक टेंसर वाली एक सूची लौटाता है।
सार्वजनिक तरीके
स्थिर XlaRecvTPUEएम्बेडिंगएक्टिवेशन | |
इटरेटर < ऑपरेंड <फ्लोट>> | पुनरावर्तक () |
सूची< आउटपुट <फ़्लोट>> | आउटपुट () मॉडल में प्रति एम्बेडिंग तालिका में एक टेंसर युक्त एम्बेडिंग सक्रियणों की एक TensorList। |
विरासत में मिली विधियाँ
सार्वजनिक तरीके
सार्वजनिक स्थैतिक XlaRecvTPUEmbeddingActivations बनाएँ ( स्कोप स्कोप, ऑपरेंड <?> डिडुप्लीकेशनडेटा, लंबी संख्याएँ, स्ट्रिंग कॉन्फ़िगरेशन)
एक नए XlaRecvTPUEmbeddingActivations ऑपरेशन को लपेटकर एक क्लास बनाने की फ़ैक्टरी विधि।
पैरामीटर
दायरा | वर्तमान दायरा |
---|---|
डिडुप्लीकेशनडेटा | प्रकार=DT_VARIANT वाला एक टेंसर जिसमें डिडुप्लीकेशन डेटा होता है। टेंसर एक XLA नेस्टेड टपल है जिसमें N तत्व होते हैं (जहाँ N प्रति TPU चिप में टेंसर कोर की एम्बेडिंग की संख्या का अनुपात है)। नेस्टेड ट्यूपल का प्रत्येक तत्व रैंक 1 टेंसर का ट्यूपल है। प्रत्येक टेंसर में या तो TensorCore पर एम्बेडिंग लुकअप के लिए इंडेक्स (DT_UINT32) या एम्बेडिंग लुकअप ऑपरेशन के आउटपुट पर लागू करने के लिए वेट (DT_FLOAT) होते हैं। |
संख्या तालिकाएँ | आउटपुट सक्रियण टेंसरों की संख्या. यदि फीचर डिस्क्रिप्टर टीपीयू एम्बेडिंग कॉन्फ़िगरेशन में मौजूद है, तो यह सुविधाओं की संख्या के बराबर है अन्यथा मॉडल में एम्बेडिंग तालिकाओं की संख्या के बराबर है। |
कॉन्फ़िग | क्रमबद्ध टीपीयूएम्बेडिंगकॉन्फिगरेशन प्रोटो। |
रिटर्न
- XlaRecvTPUEmbeddingActivations का एक नया उदाहरण
सार्वजनिक सूची < आउटपुट <फ्लोट>> आउटपुट ()
मॉडल में प्रति एम्बेडिंग तालिका में एक टेंसर युक्त एम्बेडिंग सक्रियणों की एक TensorList।