EnqueueTPUEmbeddingRaggedTensorBatch

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

يسهل عملية نقل التعليمات البرمجية التي تستخدم tf.nn.embedding_lookup().

تتوافق Sample_splits[i] وembedding_indices[i] وaggregation_weights[i] مع الميزة i. يشير table_ids[i] إلى جدول التضمين الذي سيتم البحث عن الميزة فيه.

يجب أن يكون للموترات في المواضع المقابلة في قائمتين من قوائم الإدخال، مؤشرات التضمين وأوزان التجميع، نفس الشكل، أي الرتبة 1 مع dim_size() مساوية لإجمالي عدد عمليات البحث في الجدول الموضح بواسطة الميزة المقابلة.

فئات متداخلة

فصل EnqueueTPUEmbeddingRaggedTensorBatch.Options السمات الاختيارية لـ EnqueueTPUEmbeddingRaggedTensorBatch

الأساليب العامة

ثابت EnqueueTPUEmbeddingRaggedTensorBatch.Options
المجمعات (قائمة <سلسلة> المجمعات)
ثابت <T يمتد الرقم، U يمتد الرقم، V يمتد الرقم> EnqueueTPUEmbeddingRaggedTensorBatch
إنشاء (نطاق النطاق ، Iterable< operand <T>> SampleSplits، Iterable< operand <U>> embeddingIndices، Iterable< operand <V>> aggregationWeights، Operand <String> modeOverride، List<Long> tableIds، Options... options)
طريقة المصنع لإنشاء فئة تلتف حول عملية EnqueueTPUEmbeddingRaggedTensorBatch الجديدة.
ثابت EnqueueTPUEmbeddingRaggedTensorBatch.Options
جهاز ترتيبي (جهاز ترتيبي طويل)
ثابت EnqueueTPUEmbeddingRaggedTensorBatch.Options
maxSequenceLengths (قائمة<Long> maxSequenceLengths)
ثابت EnqueueTPUEmbeddingRaggedTensorBatch.Options
عدد الميزات (قائمة <طويلة> عدد الميزات)

الطرق الموروثة

الأساليب العامة

مجمعات EnqueueTPUEmbeddingRaggedTensorBatch.Options العامة الثابتة (مجموعات List<String>)

حدود
المجمعات قائمة بمقاييس السلسلة، واحدة لكل جدول تضمين تحدد كيفية تسوية عمليات تنشيط التضمين بعد الجمع المرجح. الموحدات المدعومة هي "المتوسط" أو "المجموع" أو "sqrtn". من غير الصحيح أن يكون مجموع الأوزان 0 لـ "mean" أو أن يكون مجموع الأوزان المربعة 0 لـ "sqrtn". إذا لم يتم تمرير أدوات الدمج، فإن الإعداد الافتراضي هو استخدام "sum" لجميع الجداول.

إنشاء EnqueueTPUEmbeddingRaggedTensorBatch العام الثابت ( نطاق النطاق ، Iterable< Operand <T>> SampleSplits، Iterable< operand <U>> embeddingIndices، Iterable< operand <V>> aggregationWeights، Operand <String> modeOverride، List<Long> tableIds، Options.. . خيارات)

طريقة المصنع لإنشاء فئة تلتف حول عملية EnqueueTPUEmbeddingRaggedTensorBatch الجديدة.

حدود
نِطَاق النطاق الحالي
عينة سبليتس قائمة Tensors من المرتبة الأولى تحدد نقاط التوقف لتقسيم مؤشرات التضمين وأوزان التجميع إلى صفوف. وهو يتوافق مع ids.row_splits في embedding_lookup()، عندما تكون ids عبارة عن RaggedTensor.
embeddingIndices قائمة الموترات من المرتبة الأولى، والمؤشرات في جداول التضمين. وهو يتوافق مع ids.values ​​في embedding_lookup()، عندما تكون ids عبارة عن RaggedTensor.
aggregationWeights قائمة الموترات من الرتبة 1 تحتوي على أوزان التجميع لكل مثال تدريبي. وهو يتوافق مع حقل القيم الخاص بـ RaggedTensor مع نفس Row_splits كمعرفات في embedding_lookup()، عندما تكون ids عبارة عن RaggedTensor.
modeOverride إدخال سلسلة يتجاوز الوضع المحدد في TPUEmbeddingConfiguration. القيم المدعومة هي {'unspecified'، 'inference'، 'training'، 'backward_pass_only'}. عند التعيين على "غير محدد"، يتم استخدام الوضع المعين في TPUEmbeddingConfiguration، وإلا يتم استخدام mode_override.
معرفات الجدول قائمة من الأعداد الصحيحة تحدد معرف جدول التضمين (إزاحة TableDescriptor في TPUEmbeddingConfiguration) للبحث عن المدخلات المقابلة. يتم البحث عن الإدخال رقم باستخدام table_ids[i]. يجب أن يكون حجم قائمة table_ids مساويا لحجم Sample_indices وembedding_indices وaggregation_weights.
خيارات يحمل قيم السمات الاختيارية
عائدات
  • مثيل جديد لـ EnqueueTPUEmbeddingRaggedTensorBatch

EnqueueTPUEmbeddingRaggedTensorBatch.Options العام الثابت ، الجهازOrdinal (جهاز طويل Ordinal)

حدود
جهازترتيبي جهاز TPU للاستخدام. يجب أن يكون >= 0 وأقل من عدد نوى TPU في المهمة التي تم وضع العقدة عليها.

EnqueueTPUEmbeddingRaggedTensorBatch.Options maxSequenceLengths (قائمة<Long> maxSequenceLengths) ثابتة عامة

EnqueueTPUEmbeddingRaggedTensorBatch.Options numFeatures (قائمة <Long> numFeatures) ثابتة عامة