CrossReplicaSum

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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