ReadVariableXlaSplitND

सभी आयामों में संसाधन चर इनपुट टेंसर को विभाजित करता है।

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

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

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

[[0, 1, 2],
  [3, 4, 5],
  [6, 7, 8]]
 

[2, 2]
 

[1, 1]
 

[[0, 1],
  [3, 4]]
 [[2, 0],
  [5, 0]]
 [[6, 7],
  [0, 0]]
 [[8, 0],
  [0, 0]]