TPUPartitionedOutputV2

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

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

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

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

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

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

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

सार्वजनिक स्थैतिक TPUPartitionedOutputV2 <T> बनाएं ( स्कोप स्कोप, ऑपरेंड <T> इनपुट, लॉन्ग नंबरस्प्लिट्स, लिस्ट <लॉन्ग> पार्टीशनडिम्स)

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

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

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

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

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