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