सभी इनपुट टेंसरों में कमी युक्त एक टेंसर आउटपुट करता है।
एक टेंसर को आउटपुट करता है जिसमें समान `shared_name के भीतर ऑप्स को पास किए गए सभी इनपुट टेंसरों में कमी होती है।
ग्राफ़ का निर्माण इस प्रकार किया जाना चाहिए कि यदि एक ऑप शेयर्ड_नाम मान `सी` के साथ चलता है, तो `num_devices` ऑप्स शेयर्ड_नाम मान `सी` के साथ चलेगा। ऐसा करने में विफलता के कारण ग्राफ़ निष्पादन पूरा होने में विफल हो जाएगा।
इनपुट: कमी डेटा का इनपुट: सभी `num_devices` उपकरणों में कमी का मूल्य। कमी: प्रदर्शन करने के लिए कमी ऑपरेशन। num_devices: इस कमी में भाग लेने वाले उपकरणों की संख्या। साझा_नाम: पहचानकर्ता जो समान कमी के ऑप्स के बीच साझा किया जाता है।
सार्वजनिक तरीके
आउटपुट <T> | आउटपुट के रूप में () टेंसर का प्रतीकात्मक हैंडल लौटाता है। |
स्थिर <T संख्या बढ़ाता है> NcclAllReduce <T> | |
आउटपुट <T> | डेटा () |
विरासत में मिली विधियाँ
सार्वजनिक तरीके
सार्वजनिक आउटपुट <T> asOutput ()
टेंसर का प्रतीकात्मक हैंडल लौटाता है।
TensorFlow संचालन के इनपुट किसी अन्य TensorFlow ऑपरेशन के आउटपुट हैं। इस पद्धति का उपयोग एक प्रतीकात्मक हैंडल प्राप्त करने के लिए किया जाता है जो इनपुट की गणना का प्रतिनिधित्व करता है।
सार्वजनिक स्थैतिक NcclAllReduce <T> बनाएं ( स्कोप स्कोप, ऑपरेंड <T> इनपुट, स्ट्रिंग में कमी, लंबी संख्या डिवाइस, स्ट्रिंग साझा नाम)
एक नया NcclAllReduce ऑपरेशन लपेटकर एक क्लास बनाने की फ़ैक्टरी विधि।
पैरामीटर
दायरा | वर्तमान दायरा |
---|
रिटर्न
- NcclAllReduce का एक नया उदाहरण