SparseSegmentSqrtN के लिए ग्रेडिएंट की गणना करता है।
ग्रेड के समान आकार के साथ टेंसर "आउटपुट" लौटाता है, आयाम 0 को छोड़कर जिसका मान "सूचकांक" में अद्वितीय अनुक्रमितों की संख्या है। साथ ही वेक्टर "sorted_unique_indices" भी लौटाता है जिसमें "सूचकांक" से संबंधित सूचकांक शामिल होते हैं।
सार्वजनिक तरीके
स्थिर <T संख्या बढ़ाता है, U संख्या बढ़ाता है, V संख्या बढ़ाता है> SparseSegmentSqrtNGradV2 <T, U> | |
आउटपुट <T> | आउटपुट () |
आउटपुट <यू> |
विरासत में मिली विधियाँ
सार्वजनिक तरीके
सार्वजनिक स्थैतिक SparseSegmentSqrtNGradV2 <T, U> बनाएं ( स्कोप स्कोप, ऑपरेंड <T> ग्रेड, ऑपरेंड <U> इंडेक्स, ऑपरेंड <V> सेगमेंटआईडी, ऑपरेंड <इंटेगर>डेंसआउटपुटडिम0)
एक नए SparseSegmentSqrtNGradV2 ऑपरेशन को लपेटकर एक क्लास बनाने की फ़ैक्टरी विधि।
पैरामीटर
दायरा | वर्तमान दायरा |
---|---|
ग्रैड | ग्रेडिएंट को SparseSegmentSqrtN ऑप में प्रचारित किया गया। |
सूचकांक | सूचकांक संबंधित SparseSegmentSqrtN op को पास कर दिए गए। |
खंड आईडी | सेगमेंट_आईडी संबंधित SparseSegmentSqrtN ऑप को पास कर दिया गया। |
सघनआउटपुटडिम0 | "डेटा" का आयाम 0 SparseSegmentSqrtN op को भेजा गया। |
रिटर्न
- SparseSegmentSqrtNGradV2 का एक नया उदाहरण