পাবলিক চূড়ান্ত বর্গ CrossReplicaSum
প্রতিলিপিকৃত TPU দৃষ্টান্ত জুড়ে ইনপুট যোগ করার জন্য একটি বিকল্প।
প্রতিটি উদাহরণ তার নিজস্ব ইনপুট সরবরাহ করে।
উদাহরণস্বরূপ, ধরুন 8 টি টিপিইউ উদাহরণ রয়েছে: `[A, B, C, D, E, F, G, H]`। গ্রুপ_অ্যাসাইনমেন্ট =`[[0,2,4,6],[1,3,5,7]]` সেট করা `A, C, E, G` কে গ্রুপ 0 হিসেবে সেট করে এবং `B, D, F, H` গ্রুপ 1 হিসাবে। এভাবে আমরা আউটপুট পাই: `[A+C+E+G, B+D+F+H, A+C+E+G, B+D+F+H, A+C+E+ G, B+D+F+H, A+C+E+G, B+D+F+H]`।
পাবলিক পদ্ধতি
| আউটপুট <টি> | asOutput () একটি টেনসরের প্রতীকী হ্যান্ডেল প্রদান করে। |
| স্ট্যাটিক <টি নম্বর প্রসারিত> CrossReplicaSum <টি> | |
| আউটপুট <টি> | আউটপুট () সমস্ত বিতরণ করা ইনপুটের যোগফল। |
উত্তরাধিকারসূত্রে প্রাপ্ত পদ্ধতি
পাবলিক পদ্ধতি
পাবলিক আউটপুট <টি> asOutput ()
একটি টেনসরের প্রতীকী হ্যান্ডেল প্রদান করে।
TensorFlow অপারেশনের ইনপুট হল অন্য TensorFlow অপারেশনের আউটপুট। এই পদ্ধতিটি একটি সিম্বলিক হ্যান্ডেল পেতে ব্যবহৃত হয় যা ইনপুটের গণনার প্রতিনিধিত্ব করে।
পাবলিক স্ট্যাটিক CrossReplicaSum <টি> তৈরি করুন ( ব্যাপ্তি সুযোগ প্রতীক <টি> ইনপুট, প্রতীক <integer> groupAssignment)
একটি নতুন CrossReplicaSum অপারেশন মোড়ানো একটি ক্লাস তৈরি করার কারখানার পদ্ধতি।
পরামিতি
| সুযোগ | বর্তমান সুযোগ |
|---|---|
| ইনপুট | সমষ্টিতে স্থানীয় ইনপুট। |
| গ্রুপ অ্যাসাইনমেন্ট | আকৃতি সহ একটি int32 টেনসর [সংখ্যা_গোষ্ঠী, সংখ্যা_প্রতিরূপ_প্রতি_গ্রুপ]। `group_assignment[i]` ith সাবগ্রুপের রেপ্লিকা আইডি উপস্থাপন করে। |
রিটার্নস
- CrossReplicaSum এর একটি নতুন উদাহরণ