SparseSegmentSumGradV2

सार्वजनिक अंतिम वर्ग SparseSegmentSumGradV2

SparseSegmentSum के लिए ग्रेडिएंट की गणना करता है।

ग्रेड के समान आकार के साथ टेंसर "आउटपुट" लौटाता है, आयाम 0 को छोड़कर जिसका मान "सूचकांक" में अद्वितीय अनुक्रमितों की संख्या है। साथ ही वेक्टर "sorted_unique_indices" भी लौटाता है जिसमें "सूचकांक" से संबंधित सूचकांक शामिल होते हैं।

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

स्थिर <T संख्या बढ़ाता है, U संख्या बढ़ाता है, V संख्या बढ़ाता है> SparseSegmentSumGradV2 <T, U>
बनाएं ( स्कोप स्कोप, ऑपरेंड <टी> ग्रेड, ऑपरेंड <यू> इंडेक्स, ऑपरेंड <वी> सेगमेंटआईडी, ऑपरेंड <इंटेगर>डेंसआउटपुटडिम0)
एक नए SparseSegmentSumGradV2 ऑपरेशन को लपेटकर एक क्लास बनाने की फ़ैक्टरी विधि।
आउटपुट <T>
आउटपुट ()
आउटपुट <यू>
क्रमबद्धअनूठे सूचकांक ()

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

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

सार्वजनिक स्थैतिक SparseSegmentSumGradV2 <T, U> बनाएं ( स्कोप स्कोप, ऑपरेंड <T> ग्रेड, ऑपरेंड <U> इंडेक्स, ऑपरेंड <V> सेगमेंटआईडी, ऑपरेंड <इंटेगर>डेंसआउटपुटडिम0)

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

पैरामीटर
दायरा वर्तमान दायरा
ग्रैड ग्रेडिएंट को SparseSegmentSum ऑप में प्रचारित किया गया।
सूचकांक सूचकांक संबंधित SparseSegmentSum op को पास कर दिए गए।
खंड आईडी सेगमेंट_आईडी संबंधित स्पार्ससेगमेंटसम ऑप को पास कर दिया गया।
सघनआउटपुटडिम0 "डेटा" का आयाम 0 SparseSegmentSum op को पास कर दिया गया।
रिटर्न
  • SparseSegmentSumGradV2 का एक नया उदाहरण

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

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