सार्वजनिक अंतिम वर्ग TPUPartitionedOutputV2
एक ऑप जो एक टेंसर को डिमल्टीप्लेक्स करता है जिसे XLA द्वारा विभाजित सूची में विभाजित किया जाता है
XLA गणना के बाहर आउटपुट। एनडी शार्डिंग का समर्थन करता है।
सार्वजनिक तरीके
स्थिर <T> TPUpartitionedOutputV2 <T> | |
इटरेटर< ऑपरेंड <टी>> | पुनरावर्तक () |
सूची< आउटपुट <टी>> | आउटपुट () विभाजित आउटपुट की एक सूची जिसका आकार समान है। |
विरासत में मिली विधियाँ
सार्वजनिक तरीके
सार्वजनिक स्थैतिक TPUpartitionedOutputV2 <T> बनाएं ( स्कोप स्कोप, ऑपरेंड <T> इनपुट, लंबी संख्या विभाजन, सूची<लंबी> पार्टीशनडिम्स)
एक नया TPUpartitionedOutputV2 ऑपरेशन लपेटकर एक क्लास बनाने की फ़ैक्टरी विधि।
पैरामीटर
दायरा | वर्तमान दायरा |
---|---|
आदानों | एक टेंसर जो विभाजित टेंसरों के पूर्ण आकार का प्रतिनिधित्व करता है। |
विभाजनडिम्स | प्रत्येक आयाम को कैसे विभाजित किया जाता है, इसका वर्णन करने वाले पूर्णांकों की एक सूची। रिक्तता इंगित करती है कि इनपुट दोहराए गए हैं। |
रिटर्न
- TPUpartitionedOutputV2 का एक नया उदाहरण