কালেকটিভ পারমিউট

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

প্রতিলিপিকৃত TPU দৃষ্টান্ত জুড়ে টেনসর পারমিউট করার জন্য একটি অপশন।

প্রতিটি উদাহরণ তার নিজস্ব ইনপুট সরবরাহ করে।

উদাহরণস্বরূপ, ধরুন 4 টি টিপিইউ উদাহরণ রয়েছে: `[A, B, C, D]`। source_target_pairs=`[[0,1],[1,2],[2,3],[3,0]]` পাস করলে আউটপুট পাওয়া যায়: `[D, A, B, C]`।

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

আউটপুট <T>
আউটপুট হিসাবে ()
একটি টেনসরের প্রতীকী হ্যান্ডেল প্রদান করে।
স্ট্যাটিক <T> CollectivePermute <T>
তৈরি করুন ( স্কোপ স্কোপ, Operand <T> ইনপুট, Operand <Integer> sourceTargetPairs)
একটি নতুন CollectivePermute অপারেশন মোড়ানো একটি ক্লাস তৈরি করার কারখানার পদ্ধতি।
আউটপুট <T>
আউটপুট ()
অনুমতি দেওয়া ইনপুট।

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

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

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

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

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

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

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

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

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

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