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