सार्वजनिक अंतिम वर्ग EnqueueTPUEmbeddingIntegerBatch
एक ऑप जो इनपुट बैच टेंसरों की एक सूची को टीपीयूएम्बेडिंग में जोड़ता है।
नेस्टेड क्लासेस
कक्षा | EnqueueTPUEmbeddingIntegerBatch.Options | EnqueueTPUEmbeddingIntegerBatch के लिए वैकल्पिक विशेषताएँ |
स्थिरांक
डोरी | OP_NAME | इस ऑप का नाम, जैसा कि TensorFlow कोर इंजन द्वारा जाना जाता है |
सार्वजनिक तरीके
स्थिर EnqueueTPUEmbeddingIntegerBatch | |
स्थिर EnqueueTPUEmbeddingIntegerBatch.Options | डिवाइसऑर्डिनल (लंबा डिवाइसऑर्डिनल) |
विरासत में मिली विधियाँ
स्थिरांक
सार्वजनिक स्थैतिक अंतिम स्ट्रिंग OP_NAME
इस ऑप का नाम, जैसा कि TensorFlow कोर इंजन द्वारा जाना जाता है
स्थिर मान: "EnqueueTPUEmbeddingIntegerBatch"
सार्वजनिक तरीके
सार्वजनिक स्थैतिक EnqueueTPUEmbeddingIntegerBatch बनाएं ( स्कोप स्कोप, Iterable < ऑपरेंड < TInt32 >> बैच, ऑपरेंड < TString > modeOverride, विकल्प... विकल्प)
एक नया EnqueueTPUEmbeddingIntegerBatch ऑपरेशन लपेटकर एक क्लास बनाने की फ़ैक्टरी विधि।
पैरामीटर
दायरा | वर्तमान दायरा |
---|---|
बैच | 1डी टेंसरों की एक सूची, प्रत्येक एम्बेडिंग तालिका के लिए एक, जिसमें तालिकाओं में सूचकांक शामिल हैं। |
मोडओवरराइड | एक स्ट्रिंग इनपुट जो TPUEmbeddingConfiguration में निर्दिष्ट मोड को ओवरराइड करता है। समर्थित मान {'अनिर्दिष्ट', 'अनुमान', 'प्रशिक्षण', 'बैकवर्ड_पास_ओनली'} हैं। जब 'अनिर्दिष्ट' पर सेट किया जाता है, तो TPUEmbeddingConfiguration में सेट मोड का उपयोग किया जाता है, अन्यथा mode_override का उपयोग किया जाता है। |
विकल्प | वैकल्पिक गुण मान रखता है |
रिटर्न
- EnqueueTPUEmbeddingIntegerBatch का एक नया उदाहरण
सार्वजनिक स्थैतिक EnqueueTPUEmbeddingIntegerBatch.Options डिवाइसऑर्डिनल (लंबा डिवाइसऑर्डिनल)
पैरामीटर
डिवाइसऑर्डिनल | उपयोग करने के लिए TPU डिवाइस. >= 0 होना चाहिए और उस कार्य में टीपीयू कोर की संख्या से कम होना चाहिए जिस पर नोड रखा गया है। |
---|