ইনপুট টেনসরকে সমস্ত মাত্রা জুড়ে বিভক্ত করে।
একটি অপ যা প্রদত্ত num_splits বৈশিষ্ট্যের উপর ভিত্তি করে ইনপুট টেনসরকে স্লাইস করে, প্যাডগুলি ঐচ্ছিকভাবে স্লাইস করে এবং স্লাইসগুলি ফেরত দেয়। স্লাইস সারি প্রধান ক্রম ফেরত দেওয়া হয়.
এই অপটি টিপিইউ সেতুর মাধ্যমে তৈরি করা যেতে পারে।
উদাহরণস্বরূপ, `ইনপুট` টেনসরের সাথে:
[[0, 1, 2],
[3, 4, 5],
[6, 7, 8]]
`num_splits`: [2, 2]
এবং `paddings`: [1, 1]
প্রত্যাশিত `আউটপুট` হল: [[0, 1],
[3, 4]]
[[2, 0],
[5, 0]]
[[6, 7],
[0, 0]]
[[8, 0],
[0, 0]]
নেস্টেড ক্লাস
ক্লাস | XlaSplitND.Options | XlaSplitND এর জন্য ঐচ্ছিক বৈশিষ্ট্য |
পাবলিক পদ্ধতি
স্ট্যাটিক <T> XlaSplitND <T> | |
ইটারেটর< অপারেন্ড <T>> | |
তালিকা< আউটপুট <T>> | আউটপুট () |
স্ট্যাটিক XlaSplitND.Options | প্যাডিং (লিস্ট<লং> প্যাডিং) |
উত্তরাধিকারসূত্রে প্রাপ্ত পদ্ধতি
পাবলিক পদ্ধতি
পাবলিক স্ট্যাটিক XlaSplitND <T> তৈরি করুন ( স্কোপ স্কোপ, অপারেন্ড <T> ইনপুট, লং এন, লিস্ট<লং> নম্বর স্প্লিট, বিকল্প... বিকল্প)
একটি নতুন XlaSplitND অপারেশন মোড়ানো একটি ক্লাস তৈরি করার কারখানা পদ্ধতি।
পরামিতি
সুযোগ | বর্তমান সুযোগ |
---|---|
ইনপুট | সমস্ত মাত্রা জুড়ে বিভক্ত করার জন্য ইনপুট টেনসর। } out_arg { নাম: "outputs" বর্ণনা: < |
numSplits | প্রতি মাত্রা বিভক্ত করার উপায় সংখ্যা. আকৃতির মাত্রা অবশ্যই সমানভাবে বিভাজ্য হতে হবে। |
বিকল্প | ঐচ্ছিক বৈশিষ্ট্য মান বহন করে |
রিটার্নস
- XlaSplitND এর একটি নতুন উদাহরণ
পাবলিক স্ট্যাটিক XlaSplitND.Options প্যাডিং (List<Long> paddings)
পরামিতি
প্যাডিং | বিভক্ত করার আগে প্রয়োগ করার জন্য ইনপুট টেনসরের প্রতি মাত্রার ডান প্যাডিংয়ের ঐচ্ছিক তালিকা। এটি একটি মাত্রা সমানভাবে বিভাজ্য করতে ব্যবহার করা যেতে পারে। |
---|