सार्वजनिक अंतिम वर्ग CollectivePermute
दोहराए गए टीपीयू उदाहरणों में टेंसर को अनुमति देने के लिए एक ओप।
प्रत्येक उदाहरण अपने स्वयं के इनपुट की आपूर्ति करता है।
उदाहरण के लिए, मान लीजिए कि 4 टीपीयू इंस्टेंस हैं: `[ए, बी, सी, डी]`। पासिंग source_target_pairs=`[[0,1],[1,2],[2,3],[3,0]]` को आउटपुट मिलता है: `[D, A, B, C]`।
सार्वजनिक तरीके
| आउटपुट <टी> | asOutput () टेंसर का प्रतीकात्मक हैंडल लौटाता है। |
| स्थिर <टी> CollectivePermute <टी> | |
| आउटपुट <टी> | उत्पादन () अनुमत इनपुट। |
विरासत में मिली विधियां
सार्वजनिक तरीके
सार्वजनिक आउटपुट <टी> asOutput ()
टेंसर का प्रतीकात्मक हैंडल लौटाता है।
TensorFlow संचालन के लिए इनपुट एक अन्य TensorFlow ऑपरेशन के आउटपुट हैं। इस पद्धति का उपयोग एक प्रतीकात्मक हैंडल प्राप्त करने के लिए किया जाता है जो इनपुट की गणना का प्रतिनिधित्व करता है।
सार्वजनिक स्थिर CollectivePermute <टी> बनाने ( स्कोप गुंजाइश, ओपेरैंड <टी> इनपुट, ओपेरैंड <पूर्णांक> sourceTargetPairs)
एक नया कलेक्टिवपरम्यूट ऑपरेशन रैपिंग क्लास बनाने के लिए फ़ैक्टरी विधि।
मापदंडों
| दायरा | वर्तमान दायरा |
|---|---|
| इनपुट | स्थानीय इनपुट को परमिट किया जाना है। वर्तमान में केवल फ्लोट और bfloat16 का समर्थन करता है। |
| स्रोत लक्ष्य जोड़े | आकार के साथ एक टेंसर [num_pairs, 2]। |
रिटर्न
- कलेक्टिव परम्यूट का एक नया उदाहरण