عملية تقوم بإدراج مؤشرات إدخال TPUEmbedding من SparseTensor.
تعمل هذه العملية على تسهيل عملية نقل التعليمات البرمجية التي تستخدم embedding_lookup_sparse()، على الرغم من أن بعض عمليات المعالجة المسبقة لـ Python لوسائط SparseTensor إلى embedding_lookup_sparse() مطلوبة لإنتاج الوسائط لهذه العملية، حيث يُسمح فقط بـ EnqueueTPUEmbeddingSparseBatch Op لكل خطوة تدريب.
يجب أن يكون للموترات في المواضع المقابلة في قوائم الإدخال الثلاثة نفس الشكل، أي أن المرتبة 1 مع dim_size() تساوي العدد الإجمالي لعمليات البحث في الجدول الموصوف بواسطة table_id المقابل.
فئات متداخلة
فصل | EnqueueTPUEmbeddingSparseBatch.Options | سمات اختيارية لـ EnqueueTPUEmbeddingSparseBatch |
الأساليب العامة
ثابت EnqueueTPUEmbeddingSparseBatch.Options | المجمعات (قائمة <سلسلة> المجمعات) |
ثابت <T يمتد الرقم، U يمتد الرقم، V يمتد الرقم> EnqueueTPUEmbeddingSparseBatch | |
ثابت EnqueueTPUEmbeddingSparseBatch.Options | جهاز ترتيبي (جهاز ترتيبي طويل) |
الطرق الموروثة
الأساليب العامة
مجمعات EnqueueTPUEmbeddingSparseBatch.Options العامة الثابتة (مجموعات List<String>)
حدود
المجمعات | قائمة بمقاييس السلسلة، واحدة لكل جدول تضمين تحدد كيفية تسوية عمليات تنشيط التضمين بعد الجمع المرجح. الموحدات المدعومة هي "المتوسط" أو "المجموع" أو "sqrtn". من غير الصحيح أن يكون مجموع الأوزان 0 لـ "mean" أو أن يكون مجموع الأوزان المربعة 0 لـ "sqrtn". إذا لم يتم تمرير أدوات الدمج، فإن الإعداد الافتراضي هو استخدام "sum" لجميع الجداول. |
---|
إنشاء EnqueueTPUEmbeddingSparseBatch الثابت العام (نطاق النطاق ، Iterable< operand <T>> SampleIndices، Iterable< operand <U>> embeddingIndices، Iterable< operand <V>> aggregationWeights، Operand <String> modeOverride، Options... options)
طريقة المصنع لإنشاء فئة تلتف حول عملية EnqueueTPUEmbeddingSparseBatch الجديدة.
حدود
نِطَاق | النطاق الحالي |
---|---|
SampleIndices | قائمة Tensors من المرتبة 1 تحدد مثال التدريب والميزة التي تنتمي إليها قيم embedding_indices وaggregation_weights المقابلة. يجب أن تساوي Sample_indices[i] b * nf + f، حيث nf هو عدد الميزات من الجدول المقابل، وf موجود في [0، nf)، وb موجود في [0، حجم الدُفعة). |
embeddingIndices | قائمة الموترات من المرتبة الأولى، والمؤشرات في جداول التضمين. |
aggregationWeights | قائمة Tensors من الرتبة 1 تحتوي على كل عينة - أي لكل (مثال تدريبي، ميزة) - أوزان التجميع. |
modeOverride | إدخال سلسلة يتجاوز الوضع المحدد في TPUEmbeddingConfiguration. القيم المدعومة هي {'unspecified'، 'inference'، 'training'، 'backward_pass_only'}. عند التعيين على "غير محدد"، يتم استخدام الوضع المعين في TPUEmbeddingConfiguration، وإلا يتم استخدام mode_override. |
خيارات | يحمل قيم السمات الاختيارية |
عائدات
- مثيل جديد لـ EnqueueTPUEmbeddingSparseBatch
EnqueueTPUEmbeddingSparseBatch.Options العام الثابت ، الجهازOrdinal (جهاز طويل Ordinal)
حدود
جهازترتيبي | جهاز TPU للاستخدام. يجب أن يكون >= 0 وأقل من عدد نوى TPU في المهمة التي تم وضع العقدة عليها. |
---|