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