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