AssignVariableXlaConcatND

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

सभी आयामों में इनपुट टेंसर को जोड़ता है।

एक ऑप जो विलय करता है, दिए गए num_splits विशेषता के आधार पर इनपुट टेंसर को स्लाइस करता है, वैकल्पिक रूप से पैडिंग को स्ट्रिप करता है, और मर्ज किए गए टेंसर को बिना पैडिंग के संसाधन चर में लिखता है।

यह ऑप टीपीयू ब्रिज के माध्यम से उत्पन्न किया जा सकता है।

उदाहरण के लिए, `इनपुट` टेंसर के साथ:

[[0, 1],
  [4, 5]]
 [[2, 3],
  [6, 7]]
 [[8, 9],
  [12, 13]]
 [[10, 11],
  [14, 15]]
 
`num_splits`:
[2, 2]
 
और `पैडिंग`:
[1, 1]
 
अपेक्षित `आउटपुट` है:
[[0, 1, 2],
  [4, 5, 6],
  [8, 9, 10]]
 

नेस्टेड क्लासेस

कक्षा AssignVariableXlaConcatND.Options AssignVariableXlaConcatND के लिए वैकल्पिक विशेषताएँ

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

स्थिर <T> AssignVariableXlaConcatND
बनाएं ( स्कोप स्कोप, ऑपरेंड <?> संसाधन, पुनरावर्तनीय < ऑपरेंड <T>> इनपुट, सूची<लंबा> numConcats, विकल्प... विकल्प)
एक नए AssignVariableXlaConcatND ऑपरेशन को लपेटकर एक क्लास बनाने की फ़ैक्टरी विधि।
स्थिर AssignVariableXlaConcatND.Options
पैडिंग (सूची<लंबी> पैडिंग)

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

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

सार्वजनिक स्थैतिक AssignVariableXlaConcatND बनाएं ( स्कोप स्कोप, ऑपरेंड <?> संसाधन, Iterable< ऑपरेंड <T>> इनपुट, सूची<लंबा> numConcats, विकल्प... विकल्प)

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

पैरामीटर
दायरा वर्तमान दायरा
संसाधन सभी आयामों में संयोजित इनपुट टेंसरों के लिए संसाधन चर। } in_arg { नाम: "इनपुट" विवरण: <
numConcats प्रति आयाम मर्ज करने के तरीकों की संख्या.
विकल्प वैकल्पिक गुण मान रखता है
रिटर्न
  • AssignVariableXlaConcatND का एक नया उदाहरण

सार्वजनिक स्थैतिक AssignVariableXlaConcatND.Options पैडिंग (सूची<लंबी> पैडिंग)

पैरामीटर
गद्दी अंतिम मर्ज किए गए टेंसर से स्ट्रिप करने के लिए प्रति आयाम सही पैडिंग की वैकल्पिक सूची। ये पैडिंग पैडिंग अलग करने से पहले मर्ज किए गए परिणाम के आयाम आकार से अधिक नहीं होनी चाहिए।