পাবলিক চূড়ান্ত বর্গ SplitV
একটি টেনসরকে একটি মাত্রা বরাবর `num_split` টেনসরে বিভক্ত করে।
পাবলিক পদ্ধতি
স্ট্যাটিক <টি, ইউ প্রসারিত number> SplitV <টি> | |
Iterator < প্রতীক <টি >> | |
তালিকা < আউটপুট <টি >> | আউটপুট () টেনসর যেগুলির আকৃতি `অক্ষ` বরাবর ব্যতীত `মান` এর সাথে মেলে, যেখানে তাদের আকারগুলি `আকার_বিভক্ত[i]`। |
উত্তরাধিকারসূত্রে প্রাপ্ত পদ্ধতি
পাবলিক পদ্ধতি
পাবলিক স্ট্যাটিক SplitV <টি> তৈরি করুন ( ব্যাপ্তি সুযোগ প্রতীক <টি> মান, প্রতীক <u> sizeSplits, প্রতীক <integer> অক্ষ, লং numSplit)
একটি নতুন SplitV অপারেশন মোড়ানো একটি ক্লাস তৈরি করার কারখানার পদ্ধতি।
পরামিতি
সুযোগ | বর্তমান সুযোগ |
---|---|
মান | বিভক্ত টেনসর. |
সাইজ স্প্লিট | বিভক্ত মাত্রা বরাবর প্রতিটি আউটপুট টেনসরের মাপ ধারণকারী তালিকা। split_dim বরাবর মানের মাত্রা যোগ করতে হবে। একটি -1 থাকতে পারে যা নির্দেশ করে যে মাত্রা অনুমান করা হবে। |
অক্ষ | 0-ডি. যে মাত্রা বরাবর বিভক্ত করা হবে। অবশ্যই `[-র্যাঙ্ক(মান), র্যাঙ্ক(মান))' পরিসরে থাকতে হবে। |
রিটার্নস
- SplitV এর একটি নতুন উদাহরণ
প্রকাশ্য তালিকা < আউটপুট <টি >> আউটপুট ()
টেনসর যেগুলির আকৃতি `অক্ষ` বরাবর ব্যতীত `মান` এর সাথে মেলে, যেখানে তাদের আকারগুলি `আকার_বিভক্ত[i]`।