CollectivePermute

सार्वजनिक अंतिम वर्ग कलेक्टिवपरम्यूट

प्रतिकृति टीपीयू उदाहरणों में टेंसरों को क्रमपरिवर्तित करने का एक ऑप।

प्रत्येक उदाहरण अपना स्वयं का इनपुट प्रदान करता है।

उदाहरण के लिए, मान लीजिए कि 4 टीपीयू उदाहरण हैं: `[ए, बी, सी, डी]`। source_target_pairs=`[[0,1],[1,2],[2,3],[3,0]]` को पास करने पर आउटपुट मिलता है: `[D, A, B, C]`।

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

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

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

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

सार्वजनिक आउटपुट <T> asOutput ()

टेंसर का प्रतीकात्मक हैंडल लौटाता है।

TensorFlow संचालन के इनपुट किसी अन्य TensorFlow ऑपरेशन के आउटपुट हैं। इस पद्धति का उपयोग एक प्रतीकात्मक हैंडल प्राप्त करने के लिए किया जाता है जो इनपुट की गणना का प्रतिनिधित्व करता है।

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

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

पैरामीटर
दायरा वर्तमान दायरा
इनपुट स्थानीय इनपुट को क्रमपरिवर्तित किया जाना है. वर्तमान में केवल फ्लोट और bfloat16 का समर्थन करता है।
स्रोत लक्ष्य जोड़े आकार वाला एक टेंसर [num_pairs, 2]।
रिटर्न
  • कलेक्टिवपरम्यूट का एक नया उदाहरण

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

क्रमपरिवर्तित इनपुट.