EnqueueTPUEmbeddingSparseBatch

الفئة النهائية العامة EnqueueTPUEmbeddingSparseBatch

عملية تقوم بإدراج مؤشرات إدخال 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
إنشاء (نطاق النطاق ، Iterable< operand <T>> SampleIndices، Iterable< operand <U>> embeddingIndices، Iterable< operand <V>> aggregationWeights، Operand <String> modeOverride، Options... options)
طريقة المصنع لإنشاء فئة تلتف حول عملية 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 في المهمة التي تم وضع العقدة عليها.