EnqueueTPUEmbeddingIntegerBatch

सार्वजनिक अंतिम वर्ग EnqueueTPUEmbeddingIntegerBatch

एक ऑप जो इनपुट बैच टेंसरों की एक सूची को टीपीयूएम्बेडिंग में जोड़ता है।

नेस्टेड क्लासेस

कक्षा EnqueueTPUEmbeddingIntegerBatch.Options EnqueueTPUEmbeddingIntegerBatch के लिए वैकल्पिक विशेषताएँ

सार्वजनिक तरीके

स्थिर EnqueueTPUEmbeddingIntegerBatch
बनाएं ( स्कोप स्कोप, Iterable< ऑपरेंड <इंटेगर>> बैच, ऑपरेंड <स्ट्रिंग> मोडओवरराइड, विकल्प... विकल्प)
एक नया EnqueueTPUEmbeddingIntegerBatch ऑपरेशन लपेटकर एक क्लास बनाने की फ़ैक्टरी विधि।
स्थिर EnqueueTPUEmbeddingIntegerBatch.Options
डिवाइसऑर्डिनल (लंबा डिवाइसऑर्डिनल)

विरासत में मिली विधियाँ

सार्वजनिक तरीके

सार्वजनिक स्थैतिक EnqueueTPUEmbeddingIntegerBatch बनाएं ( स्कोप स्कोप, Iterable< ऑपरेंड <Integer>> बैच, ऑपरेंड <String> modeOverride, विकल्प... विकल्प)

एक नया EnqueueTPUEmbeddingIntegerBatch ऑपरेशन लपेटकर एक क्लास बनाने की फ़ैक्टरी विधि।

पैरामीटर
दायरा वर्तमान दायरा
बैच 1डी टेंसरों की एक सूची, प्रत्येक एम्बेडिंग तालिका के लिए एक, जिसमें तालिकाओं में सूचकांक शामिल हैं।
मोडओवरराइड एक स्ट्रिंग इनपुट जो TPUEmbeddingConfiguration में निर्दिष्ट मोड को ओवरराइड करता है। समर्थित मान {'अनिर्दिष्ट', 'अनुमान', 'प्रशिक्षण', 'बैकवर्ड_पास_ओनली'} हैं। जब 'अनिर्दिष्ट' पर सेट किया जाता है, तो TPUEmbeddingConfiguration में सेट मोड का उपयोग किया जाता है, अन्यथा mode_override का उपयोग किया जाता है।
विकल्प वैकल्पिक गुण मान रखता है
रिटर्न
  • EnqueueTPUEmbeddingIntegerBatch का एक नया उदाहरण

सार्वजनिक स्थैतिक EnqueueTPUEmbeddingIntegerBatch.Options डिवाइसऑर्डिनल (लंबा डिवाइसऑर्डिनल)

पैरामीटर
डिवाइसऑर्डिनल उपयोग करने के लिए TPU डिवाइस. >= 0 होना चाहिए और उस कार्य में टीपीयू कोर की संख्या से कम होना चाहिए जिस पर नोड रखा गया है।