EnqueueTPUEmbeddingRaggedTensorBatch

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

tf.nn.embedding_lookup() का उपयोग करने वाले कोड की पोर्टिंग को आसान बनाता है।

सैंपल_स्प्लिट्स[i], एम्बेडिंग_इंडिसेस[i] और एग्रीगेशन_वेट्स[i] ith फीचर के अनुरूप हैं। Table_ids[i] इंगित करता है कि किस एम्बेडिंग तालिका में ith सुविधा को देखना है।

दो इनपुट सूचियों, एम्बेडिंग_इंडिसेस और एग्रीगेशन_वेट्स में संबंधित स्थानों पर टेंसरों का आकार समान होना चाहिए, यानी संबंधित सुविधा द्वारा वर्णित तालिका में लुकअप की कुल संख्या के बराबर dim_size() के साथ रैंक 1 होना चाहिए।

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

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

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

स्थिर EnqueueTPUE EmbeddingRaggedTensorBatch.Options
कॉम्बिनर्स (सूची<स्ट्रिंग> कॉम्बिनर्स)
स्थिर <T संख्या बढ़ाता है, U संख्या बढ़ाता है, V संख्या बढ़ाता है> EnqueueTPUE EmbeddingRaggedTensorBatch
बनाएं ( स्कोप स्कोप, इटरेबल < ऑपरेंड <टी>> सैंपलस्प्लिट्स, इटरेबल < ऑपरेंड <यू>> एम्बेडिंग इंडेक्स, इटरेबल < ऑपरेंड <वी>> एग्रीगेशन वेट, ऑपरेंड <स्ट्रिंग> मोड ओवरराइड, लिस्ट <लॉन्ग> टेबल आईडी, विकल्प... विकल्प)
एक नया EnqueueTPUEmbeddingRaggedTensorBatch ऑपरेशन लपेटकर एक क्लास बनाने की फ़ैक्टरी विधि।
स्थिर EnqueueTPUE EmbeddingRaggedTensorBatch.Options
डिवाइसऑर्डिनल (लंबा डिवाइसऑर्डिनल)
स्थिर EnqueueTPUE EmbeddingRaggedTensorBatch.Options
maxSequenceLengths (सूची<लंबा> maxSequenceLengths)
स्थिर EnqueueTPUE EmbeddingRaggedTensorBatch.Options
संख्याविशेषताएं (सूची<लंबी>संख्याविशेषताएं)

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

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

सार्वजनिक स्थैतिक EnqueueTPUEmbeddingRaggedTensorBatch.Options संयोजक (सूची<स्ट्रिंग> संयोजक)

पैरामीटर
समेलक स्ट्रिंग स्केलर की एक सूची, प्रत्येक एम्बेडिंग तालिका के लिए एक जो निर्दिष्ट करती है कि भारित योग के बाद एम्बेडिंग सक्रियणों को कैसे सामान्य किया जाए। समर्थित संयोजक 'माध्य', 'योग', या 'sqrtn' हैं। 'माध्य' के लिए भारों का योग 0 होना या 'sqrtn' के लिए वर्ग भारों का योग 0 होना अमान्य है। यदि कॉम्बिनर्स पास नहीं हुआ है, तो डिफ़ॉल्ट सभी तालिकाओं के लिए 'योग' का उपयोग करना है।

सार्वजनिक स्थैतिक EnqueueTPUEmbeddingRaggedTensorBatch बनाएं ( स्कोप स्कोप, Iterable< ऑपरेंड <T>> नमूनास्प्लिट्स, Iterable< ऑपरेंड <U>> एम्बेडिंगइंडिसेस, Iterable< ऑपरेंड <V>> एग्रीगेशनवेट्स, ऑपरेंड <String> modeOverride, List<Long> टेबल आईडी, विकल्प.. विकल्प )

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

पैरामीटर
दायरा वर्तमान दायरा
नमूनाविभाजन एम्बेडिंग_इंडिसेस और एकत्रीकरण_वेट को पंक्तियों में विभाजित करने के लिए ब्रेक पॉइंट निर्दिष्ट करने वाले रैंक 1 टेंसर की एक सूची। यह Embedding_lookup() में ids.row_splits से मेल खाता है, जब ids एक RaggedTensor है।
एम्बेडिंगसूचकांक एम्बेडिंग तालिकाओं में रैंक 1 टेंसर, सूचकांकों की एक सूची। यह embedding_lookup() में ids.values ​​से मेल खाता है, जब ids एक RaggedTensor है।
एकत्रीकरणवजन रैंक 1 टेन्सर्स की एक सूची जिसमें प्रति प्रशिक्षण उदाहरण एकत्रीकरण भार शामिल है। यह रैग्डटेन्सर के मान फ़ील्ड से मेल खाता है जिसमें एम्बेडिंग_लुकअप() में आईडी के समान पंक्ति_स्प्लिट्स होते हैं, जब आईडी एक रैग्डटेन्सर होता है।
मोडओवरराइड एक स्ट्रिंग इनपुट जो TPUEmbeddingConfiguration में निर्दिष्ट मोड को ओवरराइड करता है। समर्थित मान {'अनिर्दिष्ट', 'अनुमान', 'प्रशिक्षण', 'बैकवर्ड_पास_ओनली'} हैं। जब 'अनिर्दिष्ट' पर सेट किया जाता है, तो TPUEmbeddingConfiguration में सेट मोड का उपयोग किया जाता है, अन्यथा mode_override का उपयोग किया जाता है।
तालिका आईडी संबंधित इनपुट को देखने के लिए एम्बेडिंग तालिका (TPUEmbeddingConfiguration में TableDescriptor का ऑफसेट) के पहचानकर्ता को निर्दिष्ट करने वाले पूर्णांकों की एक सूची। ith इनपुट को Table_ids[i] का उपयोग करके देखा जाता है। टेबल_आईडी सूची का आकार नमूना_सूचकांक, एम्बेडिंग_सूचकांक और एकत्रीकरण_वेट के बराबर होना चाहिए।
विकल्प वैकल्पिक गुण मान रखता है
रिटर्न
  • EnqueueTPUEmbeddingRaggedTensorBatch का एक नया उदाहरण

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

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

सार्वजनिक स्थैतिक EnqueueTPUE EmbeddingRaggedTensorBatch.Options maxSequenceLengths (सूची<लंबी> maxSequenceLengths)

सार्वजनिक स्थैतिक EnqueueTPUE EmbeddingRaggedTensorBatch.Options numFeatures (सूची<लंबी> numFeatures)