EnqueueTPUEmbeddingSparseTensorBatch

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

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

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

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

فئات متداخلة

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

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

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

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

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

مجمعات EnqueueTPUEmbeddingSparseTensorBatch.Options العامة الثابتة (جامعات القائمة <String>)

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

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

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

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

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

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

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

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