EnqueueTPUEmbeddingRaggedTensorBatch

کلاس نهایی عمومی EnqueueTPUEmbeddingRaggedTensorBatch

انتقال کدهایی را که از () tf.nn.embedding_lookup استفاده می کند، آسان می کند.

نمونه_شکاف[i]، شاخص_های embedding[i] و وزن_انجماد[i] با ویژگی ith مطابقت دارند. table_ids[i] نشان می‌دهد که کدام جدول تعبیه‌شده را باید جستجو کرد.

تانسورها در موقعیت های متناظر در دو لیست ورودی، embedding_indices و aggregation_weights، باید شکل یکسانی داشته باشند، یعنی رتبه 1 با dim_size() برابر با تعداد کل جستجوها در جدول توصیف شده توسط ویژگی مربوطه.

کلاس های تو در تو

کلاس EnqueueTPUEmbeddingRaggedTensorBatch.Options ویژگی های اختیاری برای EnqueueTPUEmbeddingRaggedTensorBatch

روش های عمومی

Static EnqueueTPUEmbeddingRaggedTensorBatch.Options
ترکیب کننده ها (List<String> ترکیب کننده ها)
static <T تعداد را گسترش می دهد، U تعداد را گسترش می دهد، V شماره را گسترش می دهد> EnqueueTPUEmbeddingRaggedTensorBatch
ایجاد ( Scope scope، Iterable< Operand <T>> sampleSplits، Iterable< Operand <U>> embeddingIndices، Iterable< Operand <V>> aggregationWeights، Operand <String> modeOverride، List<Long> tableIds، Options... گزینه ها)
روش کارخانه برای ایجاد کلاسی که یک عملیات EnqueueTPUEmbeddingRaggedTensorBatch جدید را بسته بندی می کند.
Static EnqueueTPUEmbeddingRaggedTensorBatch.Options
deviceOrdinal (Long deviceOrdinal)
Static EnqueueTPUEmbeddingRaggedTensorBatch.Options
maxSequenceLengths (List<Long> maxSequenceLengths)
Static EnqueueTPUEmbeddingRaggedTensorBatch.Options
numFeatures (List<Long> numFeatures)

روش های ارثی

روش های عمومی

عمومی استاتیک EnqueueTPUEmbeddingRaggedTensorBatch.Options ترکیب کننده ها (List<String> ترکیب کننده)

مولفه های
ترکیب کننده ها فهرستی از اسکالرهای رشته ای، یکی برای هر جدول جاسازی که نحوه عادی سازی فعال سازی های جاسازی را پس از جمع وزنی مشخص می کند. ترکیب کننده های پشتیبانی شده 'mean'، 'sum' یا 'sqrtn' هستند. اینکه مجموع اوزان برای «میانگین» 0 باشد یا مجموع اوزان مجذور برای «sqrtn» 0 باشد، نامعتبر است. اگر ترکیب‌کننده‌ها تصویب نشد، پیش‌فرض استفاده از «جمع» برای همه جداول است.

عمومی static EnqueueTPUEmbeddingRaggedTensorBatch ایجاد ( محدوده دامنه، تکرارپذیر< عملوند <T>> نمونه تقسیم، تکرارپذیر< عملوند <U>> embeddingIndices، تکرارپذیر< عملوند <V>> aggregationWeights، عملوند <String> modeOverride، List<Long> جدول. . گزینه ها)

روش کارخانه برای ایجاد کلاسی که یک عملیات EnqueueTPUEmbeddingRaggedTensorBatch جدید را بسته بندی می کند.

مولفه های
محدوده محدوده فعلی
نمونه تقسیم فهرستی از تانسورهای رتبه 1 که نقاط شکست را برای تقسیم کردن embedding_indices و aggregation_weights به ردیف ها مشخص می کند. این مربوط به ids.row_splits در embedding_lookup()، زمانی که ids یک RaggedTensor است.
تعبیه شاخص ها فهرستی از تانسورهای رتبه 1، شاخص ها در جداول تعبیه شده. زمانی که id یک RaggedTensor باشد، با ids.values ​​در embedding_lookup() مطابقت دارد.
تجمع وزن ها فهرستی از تانسورهای رتبه 1 حاوی وزن‌های تجمعی در هر نمونه تمرین. این مربوط به فیلد مقادیر یک RaggedTensor با همان row_splits به عنوان id در embedding_lookup()، زمانی که ids یک RaggedTensor است.
modeOverride ورودی رشته ای که حالت مشخص شده در TPUEmbeddingConfiguration را لغو می کند. مقادیر پشتیبانی شده عبارتند از: {'نامشخص'، 'استنتاج'، 'training'، 'backward_pass_only'}. وقتی روی "نا مشخص" تنظیم شود، حالت تنظیم شده در TPUEmbeddingConfiguration استفاده می شود، در غیر این صورت mode_override استفاده می شود.
شناسه های جدول لیستی از اعداد صحیح که شناسه جدول جاسازی را مشخص می کند (تغییر جدول توصیفگر در TPUEmbeddingConfiguration) برای جستجوی ورودی مربوطه. ورودی ith با استفاده از table_ids[i] جستجو می‌شود. اندازه فهرست table_ids باید با اندازه نمونه_شاخص ها، embedding_indices و aggregation_weights برابر باشد.
گزینه ها مقادیر ویژگی های اختیاری را حمل می کند
برمی گرداند
  • یک نمونه جدید از EnqueueTPUEmbeddingRaggedTensorBatch

عمومی ثابت EnqueueTPUEmbeddingRaggedTensorBatch.Options deviceOrdinal (Long deviceOrdinal)

مولفه های
دستگاه عادی دستگاه TPU برای استفاده. باید >= 0 و کمتر از تعداد هسته های TPU در کاری که گره روی آن قرار می گیرد باشد.

عمومی ثابت EnqueueTPUEmbeddingRaggedTensorBatch.Options maxSequenceLengths (List<Long> maxSequenceLengths)

عمومی استاتیک EnqueueTPUEmbeddingRaggedTensorBatch.Options numFeatures (List<Long> numFeatures)