XlaSplitND

ইনপুট টেনসরকে সমস্ত মাত্রা জুড়ে বিভক্ত করে।

একটি অপ যা প্রদত্ত num_splits বৈশিষ্ট্যের উপর ভিত্তি করে ইনপুট টেনসরকে স্লাইস করে, প্যাডগুলি ঐচ্ছিকভাবে স্লাইস করে এবং স্লাইসগুলি ফেরত দেয়। স্লাইস সারি প্রধান ক্রম ফেরত দেওয়া হয়.

এই অপটি টিপিইউ সেতুর মাধ্যমে তৈরি করা যেতে পারে।

উদাহরণস্বরূপ, `ইনপুট` টেনসর সহ:

[[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]]