सार्वजनिक अंतिम वर्ग RecvTPUEmbeddingActivations
एक ऑप जो टीपीयू पर एम्बेडिंग सक्रियण प्राप्त करता है।
टीपीयू सिस्टम टीपीयूएम्बेडिंगएनक्यू(इंटेगर/स्पार्स/स्पार्सटेन्सर)बैच के तर्कों द्वारा निर्दिष्ट एम्बेडिंग लुकअप और एकत्रीकरण करता है। इन एकत्रीकरणों के परिणाम टेन्सरफ़्लो ग्राफ़ पर RecvTPUEmbeddingActivations ऑप के आउटपुट के रूप में दिखाई देते हैं। यह ऑप मॉडल में निर्दिष्ट प्रति तालिका सक्रियणों के एक टेंसर वाली एक सूची लौटाता है। TPU ग्राफ़ में अधिकतम एक RecvTPUEmbeddingActivations ऑप हो सकता है।
सार्वजनिक तरीके
| स्थैतिक RecvTPUEmbeddingActivations | |
| इटरेटर < ऑपरेंड <फ्लोट>> | पुनरावर्तक () |
| सूची< आउटपुट <फ़्लोट>> | आउटपुट () मॉडल में प्रति एम्बेडिंग तालिका में एक टेंसर युक्त एम्बेडिंग सक्रियणों की एक TensorList। |
विरासत में मिली विधियाँ
सार्वजनिक तरीके
सार्वजनिक स्थैतिक RecvTPUEmbeddingActivations बनाएं ( स्कोप स्कोप, लंबे numOutputs, स्ट्रिंग कॉन्फ़िगरेशन)
एक नए RecvTPUEmbeddingActivations ऑपरेशन को लपेटकर एक क्लास बनाने की फ़ैक्टरी विधि।
पैरामीटर
| दायरा | वर्तमान दायरा |
|---|---|
| numOutputs | आउटपुट सक्रियण टेंसरों की संख्या, मॉडल में एम्बेडिंग तालिकाओं की संख्या के बराबर। |
| कॉन्फ़िग | क्रमबद्ध टीपीयूएम्बेडिंगकॉन्फिगरेशन प्रोटो। |
रिटर्न
- RecvTPUEmbeddingActivations का एक नया उदाहरण
सार्वजनिक सूची < आउटपुट <फ्लोट>> आउटपुट ()
मॉडल में प्रति एम्बेडिंग तालिका में एक टेंसर युक्त एम्बेडिंग सक्रियणों की एक TensorList।