প্রতিলিপিকৃত TPU দৃষ্টান্ত জুড়ে টেনসর পারমিউট করার জন্য একটি অপশন।
প্রতিটি উদাহরণ তার নিজস্ব ইনপুট সরবরাহ করে।
উদাহরণস্বরূপ, ধরুন 4 টি টিপিইউ উদাহরণ রয়েছে: `[A, B, C, D]`। source_target_pairs=`[[0,1],[1,2],[2,3],[3,0]]` পাস করলে আউটপুট পাওয়া যায়: `[D, A, B, C]`।
ধ্রুবক
স্ট্রিং | OP_NAME | এই অপের নাম, টেনসরফ্লো কোর ইঞ্জিন দ্বারা পরিচিত |
পাবলিক পদ্ধতি
আউটপুট <T> | আউটপুট হিসাবে () টেনসরের প্রতীকী হ্যান্ডেল ফেরত দেয়। |
static <T TType প্রসারিত করে > CollectivePermute <T> | |
আউটপুট <T> | আউটপুট () অনুমতি দেওয়া ইনপুট। |
উত্তরাধিকারসূত্রে প্রাপ্ত পদ্ধতি
ধ্রুবক
সর্বজনীন স্ট্যাটিক চূড়ান্ত স্ট্রিং OP_NAME
এই অপের নাম, টেনসরফ্লো কোর ইঞ্জিন দ্বারা পরিচিত
পাবলিক পদ্ধতি
সর্বজনীন আউটপুট <T> হিসাবে আউটপুট ()
টেনসরের প্রতীকী হ্যান্ডেল ফেরত দেয়।
TensorFlow অপারেশনের ইনপুট হল অন্য TensorFlow অপারেশনের আউটপুট। এই পদ্ধতিটি একটি প্রতীকী হ্যান্ডেল পেতে ব্যবহৃত হয় যা ইনপুটের গণনাকে প্রতিনিধিত্ব করে।
পাবলিক স্ট্যাটিক CollectivePermute <T> তৈরি করুন ( Scope scope, Operand <T> ইনপুট, Operand < TInt32 > sourceTargetPairs)
একটি নতুন CollectivePermute অপারেশন মোড়ানো একটি ক্লাস তৈরি করার কারখানার পদ্ধতি।
পরামিতি
সুযোগ | বর্তমান সুযোগ |
---|---|
ইনপুট | স্থানীয় ইনপুট অনুমতি দিতে হবে। বর্তমানে শুধুমাত্র float এবং bfloat16 সমর্থন করে। |
sourceTargetPairs | আকৃতি সহ একটি টেনসর [সংখ্যা_জোড়া, 2]। |
রিটার্নস
- CollectivePermute এর একটি নতুন উদাহরণ