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