کلاس نهایی عمومی 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 در کاری که گره روی آن قرار می گیرد باشد. |
|---|