EnqueueTPUEmbeddingSparseBatch

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

عملیاتی که شاخص های ورودی TPUEmbedding را از SparseTensor در صف قرار می دهد.

این Op انتقال کدهایی را که از embedding_lookup_sparse() استفاده می‌کند، آسان می‌کند، اگرچه پیش‌پردازش پایتون از آرگومان‌های SparseTensor به embedding_lookup_sparse() برای تولید آرگومان‌های این Op مورد نیاز است، زیرا تنها یک EnqueueTPUEmbeddingSparseBatch در هر مرحله آموزشی مجاز است.

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

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

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

ثابت ها

رشته OP_NAME نام این عملیات، همانطور که توسط موتور هسته TensorFlow شناخته می شود

روش های عمومی

استاتیک EnqueueTPUEmbeddingSparseBatch.Options
ترکیب کننده ها (List<String> ترکیب کننده ها)
static <T TNumber را گسترش می دهد، U TNumber را گسترش می دهد، V TNumber را گسترش می دهد > EnqueueTPUEmbeddingSparseBatch
ایجاد ( scope scope، Iterable< Operand <T>> sampleIndices، Iterable< Operand <U>> embeddingIndices، Iterable< Operand <V>> aggregationWeights، Operand < TString > modeOverride، Options... گزینه ها)
روش کارخانه برای ایجاد کلاسی که یک عملیات EnqueueTPUEmbeddingSparseBatch جدید را بسته بندی می کند.
استاتیک EnqueueTPUEmbeddingSparseBatch.Options
deviceOrdinal (Long deviceOrdinal)

روش های ارثی

ثابت ها

رشته نهایی ثابت عمومی OP_NAME

نام این عملیات، همانطور که توسط موتور هسته TensorFlow شناخته می شود

مقدار ثابت: "EnqueueTPUEmbeddingSparseBatch"

روش های عمومی

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

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

عمومی استاتیک EnqueueTPUEmbeddingSparseBatch ایجاد ( محدوده دامنه، تکرارپذیر< عملوند <T>> نمونه شاخص، تکرارپذیر< عملوند <U>> شاخصهای embedding، تکرارپذیر< عملوند <V>> انبوه وزنها، عملوند < TString > modeOverride، گزینه‌ها... گزینه‌ها)

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

مولفه های
محدوده محدوده فعلی
شاخص های نمونه فهرستی از تانسورهای رتبه 1 که نمونه آموزشی و ویژگی را مشخص می کند که مقادیر embedding_indeces و aggregation_weights مربوط به آن است. نمونه_شاخص‌ها[i] باید برابر با b * nf + f باشد، که در آن nf تعداد ویژگی‌های جدول مربوطه است، f در [0، nf) و b در [0، اندازه دسته‌ای است).
تعبیه شاخص ها فهرستی از تانسورهای رتبه 1، شاخص ها در جداول تعبیه شده.
تجمع وزن ها فهرستی از تانسورهای رتبه 1 که در هر نمونه -- به عنوان مثال در هر (مثال آموزشی، ویژگی) - وزن های تجمعی وجود دارد.
modeOverride ورودی رشته ای که حالت مشخص شده در TPUEmbeddingConfiguration را لغو می کند. مقادیر پشتیبانی شده عبارتند از: {'نامشخص'، 'استنتاج'، 'training'، 'backward_pass_only'}. وقتی روی "نا مشخص" تنظیم شود، حالت تنظیم شده در TPUEmbeddingConfiguration استفاده می شود، در غیر این صورت mode_override استفاده می شود.
گزینه ها مقادیر ویژگی های اختیاری را حمل می کند
برمی گرداند
  • یک نمونه جدید از EnqueueTPUEmbeddingSparseBatch

عمومی ثابت EnqueueTPUEmbeddingSparseBatch.Options deviceOrdinal (طولانی دستگاهOrdinal)

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