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