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