انتقال کدهایی را که از ()tf.nn.embedding_lookup_sparse استفاده می کند، آسان می کند.
نمونه_شاخص[i]، embedding_indices[i] و aggregation_weights[i] مربوط به ویژگی ith است. table_ids[i] نشان میدهد که کدام جدول تعبیهشده را باید جستجو کرد.
تانسورها در موقعیتهای متناظر در سه فهرست ورودی (شامل_شاخصها، شاخصهای_جاسازی و وزنهای تجمع) باید شکل یکسانی داشته باشند، یعنی رتبه 1 با dim_size() برابر با تعداد کل جستجوها در جدول توصیفشده توسط ویژگی مربوطه.
کلاس های تو در تو
کلاس | EnqueueTPUEmbeddingSparseTensorBatch.Options | ویژگی های اختیاری برای EnqueueTPUEmbeddingSparseTensorBatch |
روش های عمومی
Static EnqueueTPUEmbeddingSparseTensorBatch.Options | ترکیب کننده ها (List<String> ترکیب کننده ها) |
static <T عدد را گسترش می دهد، U عدد را گسترش می دهد، V عدد را گسترش می دهد> EnqueueTPUEmbeddingSparseTensorBatch | ایجاد ( Scope scope، Iterable< Operand <T>> sampleIndices، Iterable< Operand <U>> embeddingIndices، Iterable< Operand <V>> aggregationWeights، Operand <String> modeOverride، List<Long> tableIds، Options... گزینه ها) روش کارخانه برای ایجاد کلاسی که یک عملیات EnqueueTPUEmbeddingSparseTensorBatch جدید را بسته بندی می کند. |
Static EnqueueTPUEmbeddingSparseTensorBatch.Options | deviceOrdinal (Long deviceOrdinal) |
Static EnqueueTPUEmbeddingSparseTensorBatch.Options | maxSequenceLengths (List<Long> maxSequenceLengths) |
Static EnqueueTPUEmbeddingSparseTensorBatch.Options | numFeatures (List<Long> numFeatures) |
روش های ارثی
روش های عمومی
ترکیب کننده های عمومی استاتیک EnqueueTPUEmbeddingSparseTensorBatch.Options (ترکیب کننده های List<String>)
مولفه های
ترکیب کننده ها | فهرستی از اسکالرهای رشته ای، یکی برای هر جدول جاسازی که نحوه عادی سازی فعال سازی های جاسازی را پس از جمع وزنی مشخص می کند. ترکیب کننده های پشتیبانی شده 'mean'، 'sum' یا 'sqrtn' هستند. اینکه مجموع اوزان برای «میانگین» 0 باشد یا مجموع اوزان مجذور برای «sqrtn» 0 باشد، نامعتبر است. اگر ترکیبکنندهها تصویب نشد، پیشفرض استفاده از «جمع» برای همه جداول است. |
---|
عمومی ایستا EnqueueTPUEmbeddingSparseTensorBatch ایجاد ( محدوده دامنه ، تکرارپذیر< عملوند <T>> نمونه شاخص، تکرارپذیر< عملوند <U>> شاخصهای embedding، تکرارپذیر< عملوند <V>> انبوه وزنها، عملوند <رشته> modeOverride، فهرست <طولانی> جدولهای I. . گزینه ها)
روش کارخانه برای ایجاد کلاسی که یک عملیات EnqueueTPUEmbeddingSparseTensorBatch جدید را بسته بندی می کند.
مولفه های
محدوده | محدوده فعلی |
---|---|
شاخص های نمونه | فهرستی از تانسورهای رتبه 1 که نمونه آموزشی را مشخص می کند که مقادیر embedding_indices و aggregation_weights مربوط به آن است. مطابق با sp_ids.indices[:,0] در embedding_lookup_sparse(). |
تعبیه شاخص ها | فهرستی از تانسورهای رتبه 1، شاخص ها در جداول تعبیه شده. با sp_ids.values در embedding_lookup_sparse () مطابقت دارد. |
تجمع وزن ها | فهرستی از تانسورهای رتبه 1 حاوی وزنهای تجمعی در هر نمونه تمرین. مطابق با sp_weights.values در embedding_lookup_sparse(). |
modeOverride | ورودی رشته ای که حالت مشخص شده در TPUEmbeddingConfiguration را لغو می کند. مقادیر پشتیبانی شده عبارتند از: {'نامشخص'، 'استنتاج'، 'training'، 'backward_pass_only'}. وقتی روی "نا مشخص" تنظیم شود، حالت تنظیم شده در TPUEmbeddingConfiguration استفاده می شود، در غیر این صورت mode_override استفاده می شود. |
شناسه های جدول | لیستی از اعداد صحیح که شناسه جدول جاسازی را مشخص می کند (تغییر جدول توصیفگر در TPUEmbeddingConfiguration) برای جستجوی ورودی مربوطه. ورودی ith با استفاده از table_ids[i] جستجو میشود. اندازه فهرست table_ids باید با اندازه نمونه_شاخص ها، embedding_indices و aggregation_weights برابر باشد. |
گزینه ها | مقادیر ویژگی های اختیاری را حمل می کند |
برمی گرداند
- یک نمونه جدید از EnqueueTPUEmbeddingSparseTensorBatch
عمومی استاتیک EnqueueTPUEmbeddingSparseTensorBatch.Options deviceOrdinal (Long deviceOrdinal)
مولفه های
دستگاه عادی | دستگاه TPU برای استفاده. باید >= 0 و کمتر از تعداد هسته های TPU در کاری که گره روی آن قرار می گیرد باشد. |
---|