AssignVariableXlaConcatND

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

सभी आयामों में इनपुट टेंसर को सम्मिलित करता है।

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

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

उदाहरण के लिए, `input` टेन्सर साथ:

[[0, 1],
  [4, 5]]
 [[2, 3],
  [6, 7]]
 [[8, 9],
  [12, 13]]
 [[10, 11],
  [14, 15]]
 
` num_splits`:
[2, 2]
 
और `paddings`:
[1, 1]
 
उम्मीद` outputs` है:
[[0, 1, 2],
  [4, 5, 6],
  [8, 9, 10]]
 

नेस्टेड कक्षाएं

कक्षा निरुपित चरXlaConcatND.Options के लिए वैकल्पिक विशेषताओं AssignVariableXlaConcatND

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

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

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

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

सार्वजनिक स्थिर AssignVariableXlaConcatND बनाने ( स्कोप गुंजाइश, ओपेरैंड <?> संसाधन, Iterable < ओपेरैंड <टी >> आदानों, सूची <लांग> numConcats, विकल्प ... विकल्प)

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

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

सार्वजनिक स्थिर AssignVariableXlaConcatND.Options paddings (सूची <लांग> paddings)

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