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