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