सभी आयामों में इनपुट टेंसर को जोड़ता है।
एक ऑप जो विलय करता है, दिए गए num_splits विशेषता के आधार पर इनपुट टेंसर को स्लाइस करता है, वैकल्पिक रूप से पैडिंग को स्ट्रिप करता है, और मर्ज किए गए टेंसर को बिना पैडिंग के संसाधन चर में लिखता है।
यह ऑप टीपीयू ब्रिज के माध्यम से उत्पन्न किया जा सकता है।
उदाहरण के लिए, `इनपुट` टेंसर के साथ:
[[0, 1],
[4, 5]]
[[2, 3],
[6, 7]]
[[8, 9],
[12, 13]]
[[10, 11],
[14, 15]]
[2, 2]
[1, 1]
[[0, 1, 2],
[4, 5, 6],
[8, 9, 10]]
नेस्टेड क्लासेस
कक्षा | AssignVariableXlaConcatND.Options | AssignVariableXlaConcatND के लिए वैकल्पिक विशेषताएँ |
सार्वजनिक तरीके
स्थिर <T> AssignVariableXlaConcatND | |
स्थिर AssignVariableXlaConcatND.Options | पैडिंग (सूची<लंबी> पैडिंग) |
विरासत में मिले तरीके
सार्वजनिक तरीके
सार्वजनिक स्थैतिक AssignVariableXlaConcatND बनाएं ( स्कोप स्कोप, ऑपरेंड <?> संसाधन, Iterable< ऑपरेंड <T>> इनपुट, सूची<लंबा> numConcats, विकल्प... विकल्प)
एक नए AssignVariableXlaConcatND ऑपरेशन को लपेटकर एक क्लास बनाने की फ़ैक्टरी विधि।
पैरामीटर
दायरा | वर्तमान दायरा |
---|---|
संसाधन | सभी आयामों में संयोजित इनपुट टेंसरों के लिए संसाधन चर। } in_arg { नाम: "इनपुट" विवरण: < |
numConcats | प्रति आयाम मर्ज करने के तरीकों की संख्या. |
विकल्प | वैकल्पिक गुण मान रखता है |
रिटर्न
- AssignVariableXlaConcatND का एक नया उदाहरण
सार्वजनिक स्थैतिक AssignVariableXlaConcatND.Options पैडिंग (सूची<लंबी> पैडिंग)
पैरामीटर
गद्दी | अंतिम मर्ज किए गए टेंसर से स्ट्रिप करने के लिए प्रति आयाम सही पैडिंग की वैकल्पिक सूची। ये पैडिंग पैडिंग अलग करने से पहले मर्ज किए गए परिणाम के आयाम आकार से अधिक नहीं होनी चाहिए। |
---|