CrossReplicaSum

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

प्रतिकृति टीपीयू उदाहरणों में इनपुट को जोड़ने का एक ऑप।

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

उदाहरण के लिए, मान लीजिए कि 8 टीपीयू उदाहरण हैं: `[ए, बी, सी, डी, ई, एफ, जी, एच]`। Group_assignment=`[[0,2,4,6],[1,3,5,7]]` को पास करने पर `A, C, E, G` को ग्रुप 0 और `B, D, F, H` के रूप में सेट किया जाता है। समूह 1 के रूप में। इस प्रकार हमें आउटपुट मिलते हैं: `[ए+सी+ई+जी, बी+डी+एफ+एच, ए+सी+ई+जी, बी+डी+एफ+एच, ए+सी+ई+ जी, बी+डी+एफ+एच, ए+सी+ई+जी, बी+डी+एफ+एच]`।

स्थिरांक

डोरी OP_NAME इस ऑप का नाम, जैसा कि TensorFlow कोर इंजन द्वारा जाना जाता है

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

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

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

स्थिरांक

सार्वजनिक स्थैतिक अंतिम स्ट्रिंग OP_NAME

इस ऑप का नाम, जैसा कि TensorFlow कोर इंजन द्वारा जाना जाता है

स्थिर मान: "क्रॉसरेप्लिकासम"

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

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

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

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

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

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

पैरामीटर
दायरा वर्तमान दायरा
इनपुट योग के लिए स्थानीय इनपुट.
समूह कार्य [num_groups, num_replications_per_group] आकार वाला एक int32 टेंसर। `group_assignment[i]` ith उपसमूह में प्रतिकृति आईडी का प्रतिनिधित्व करता है।
रिटर्न
  • क्रॉसरेप्लिकासम का एक नया उदाहरण

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

सभी वितरित इनपुट का योग.