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