TPUPartitionedOutputV2

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

एक ऑप जो एक टेंसर को डिमल्टीप्लेक्स करता है जिसे XLA द्वारा विभाजित सूची में विभाजित किया जाता है

XLA गणना के बाहर आउटपुट। एनडी शार्डिंग का समर्थन करता है।

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

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

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

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

सार्वजनिक स्थैतिक TPUpartitionedOutputV2 <T> बनाएं ( स्कोप स्कोप, ऑपरेंड <T> इनपुट, लंबी संख्या विभाजन, सूची<लंबी> पार्टीशनडिम्स)

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

पैरामीटर
दायरा वर्तमान दायरा
आदानों एक टेंसर जो विभाजित टेंसरों के पूर्ण आकार का प्रतिनिधित्व करता है।
विभाजनडिम्स प्रत्येक आयाम को कैसे विभाजित किया जाता है, इसका वर्णन करने वाले पूर्णांकों की एक सूची। रिक्तता इंगित करती है कि इनपुट दोहराए गए हैं।
रिटर्न
  • TPUpartitionedOutputV2 का एक नया उदाहरण

सार्वजनिक इटरेटर< ऑपरेंड <T>> इटरेटर ()

सार्वजनिक सूची < आउटपुट <टी>> आउटपुट ()

विभाजित आउटपुट की एक सूची जिसका आकार समान है।