CollectivePermute

পাবলিক চূড়ান্ত ক্লাস CollectivePermute

প্রতিলিপিকৃত 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>
তৈরি করুন ( স্কোপ স্কোপ, Operand <T> ইনপুট, Operand < TInt32 > sourceTargetPairs)
একটি নতুন CollectivePermute অপারেশন মোড়ানো একটি ক্লাস তৈরি করার কারখানার পদ্ধতি।
আউটপুট <T>
আউটপুট ()
অনুমতি দেওয়া ইনপুট।

উত্তরাধিকারসূত্রে প্রাপ্ত পদ্ধতি

ধ্রুবক

সর্বজনীন স্ট্যাটিক চূড়ান্ত স্ট্রিং OP_NAME

এই অপের নাম, টেনসরফ্লো কোর ইঞ্জিন দ্বারা পরিচিত

ধ্রুবক মান: "সম্মিলিত পারমিউট"

পাবলিক পদ্ধতি

সর্বজনীন আউটপুট <T> হিসাবে আউটপুট ()

টেনসরের প্রতীকী হ্যান্ডেল ফেরত দেয়।

TensorFlow অপারেশনের ইনপুট হল অন্য TensorFlow অপারেশনের আউটপুট। এই পদ্ধতিটি একটি প্রতীকী হ্যান্ডেল পেতে ব্যবহৃত হয় যা ইনপুটের গণনাকে প্রতিনিধিত্ব করে।

পাবলিক স্ট্যাটিক CollectivePermute <T> তৈরি করুন ( Scope scope, Operand <T> ইনপুট, Operand < TInt32 > sourceTargetPairs)

একটি নতুন CollectivePermute অপারেশন মোড়ানো একটি ক্লাস তৈরি করার কারখানার পদ্ধতি।

পরামিতি
সুযোগ বর্তমান সুযোগ
ইনপুট স্থানীয় ইনপুট অনুমতি দিতে হবে। বর্তমানে শুধুমাত্র float এবং bfloat16 সমর্থন করে।
sourceTargetPairs আকৃতি সহ একটি টেনসর [সংখ্যা_জোড়া, 2]।
রিটার্নস
  • CollectivePermute এর একটি নতুন উদাহরণ

সর্বজনীন আউটপুট <T> আউটপুট ()

অনুমতি দেওয়া ইনপুট।