EnqueueTPUEmbeddingIntegerBatch

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

عملیاتی که لیستی از تانسورهای دسته ای ورودی را در TPUEmbedding قرار می دهد.

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

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

روش های عمومی

استاتیک EnqueueTPUEmbeddingIntegerBatch
ایجاد ( دامنه دامنه ، تکرارپذیر< Operand <Integer>> دسته، عملوند <String> modeOverride، گزینه‌ها... گزینه‌ها)
روش کارخانه برای ایجاد کلاسی که یک عملیات EnqueueTPUEmbeddingIntegerBatch جدید را بسته بندی می کند.
استاتیک EnqueueTPUEmbeddingIntegerBatch.Options
deviceOrdinal (Long deviceOrdinal)

روش های ارثی

روش های عمومی

عمومی ایستا EnqueueTPUEmbeddingIntegerBatch ایجاد ( دامنه دامنه ، تکرارپذیر< Operand <Integer>> دسته، عملوند <String> modeOverride، Options... گزینه ها)

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

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

عمومی استاتیک EnqueueTPUEmbeddingIntegerBatch.Options deviceOrdinal (Long deviceOrdinal)

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