EnqueueTPUEmbeddingArbitraryTensorBatch

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

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

embedding_indices[i] و aggregation_weights[i] با ویژگی ith مطابقت دارند.

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

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

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

روش های عمومی

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

روش های ارثی

روش های عمومی

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

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

عمومی ایستا EnqueueTPUEmbeddingArbitraryTensorBatch ایجاد ( دامنه دامنه ، تکرارپذیر< Operand <T>> نمونه IndicesOrRowSplits، تکرارپذیر< عملوند <U>> embeddingIndices، تکرارپذیر< عملوند <V>> aggregationWeights، عملوند <String> Options Override,

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

مولفه های
محدوده محدوده فعلی
نمونه IndicesOrRowSplits فهرستی از تانسورهای رتبه 2 که نمونه آموزشی را مشخص می کند که مقادیر embedding_indices و aggregation_weights مربوط به آن است. اگر اندازه بعد اول آن 0 باشد، فرض می کنیم هر embedding_indices متعلق به یک نمونه متفاوت است. هر دو int32 و int64 مجاز هستند و به صورت داخلی به int32 تبدیل می شوند.

یا فهرستی از تانسورهای رتبه 1 که تقسیم‌بندی ردیف‌ها را برای تقسیم کردن embedding_indices و aggregation_weights به ردیف‌ها مشخص می‌کند. این مربوط به ids.row_splits در embedding_lookup()، زمانی که ids یک RaggedTensor است. هنگامی که تانسور ژنده‌دار ND را در نوبت قرار می‌دهیم، فقط آخرین بعد مجاز است که پاره شود. شکاف ردیف تانسور متراکم 1 بعدی است. وقتی خالی باشد، فرض می‌کنیم که یک تانسور متراکم به عملیات انتقال داده شده است. int32 و int64 هر دو مجاز هستند و در داخل به int32 تبدیل می‌شوند.

تعبیه شاخص ها فهرستی از تانسورهای رتبه 1، شاخص ها در جداول تعبیه شده. هر دو int32 و int64 مجاز هستند و به صورت داخلی به int32 تبدیل می شوند.
تجمع وزن ها فهرستی از تانسورهای رتبه 1 حاوی وزن‌های تجمعی در هر نمونه تمرین. هر دو float32 و float64 مجاز هستند و به صورت داخلی به float32 تبدیل خواهند شد.
modeOverride ورودی رشته ای که حالت مشخص شده در TPUEmbeddingConfiguration را لغو می کند. مقادیر پشتیبانی شده عبارتند از: {'نامشخص'، 'استنتاج'، 'training'، 'backward_pass_only'}. وقتی روی "نا مشخص" تنظیم شود، حالت تنظیم شده در TPUEmbeddingConfiguration استفاده می شود، در غیر این صورت mode_override استفاده می شود.
گزینه ها مقادیر ویژگی های اختیاری را حمل می کند
برمی گرداند
  • یک نمونه جدید از EnqueueTPUEmbeddingArbitraryTensorBatch

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

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