يسهل عملية نقل التعليمات البرمجية التي تستخدم 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 | |
ثابت 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 في المهمة التي تم وضع العقدة عليها. |
---|