کلاس نهایی عمومی EnqueueTPUEmbeddingBatch
عملیاتی که لیستی از تانسورهای دسته ای ورودی را در TPUEmbedding قرار می دهد.
عملیاتی که لیستی از تانسورهای دسته ای ورودی را در TPUEmbedding قرار می دهد.
کلاس های تو در تو
کلاس | EnqueueTPUEmbeddingBatch.Options | ویژگی های اختیاری برای EnqueueTPUEmbeddingBatch |
روش های عمومی
استاتیک EnqueueTPUEmbeddingBatch.Options | ترکیب کننده ها (List<String> ترکیب کننده ها) |
استاتیک EnqueueTPUEmbeddingBatch | ایجاد (حوزه دامنه ، تکرارپذیر< Operand <String>> دسته، عملوند <String> modeOverride، گزینهها... گزینهها) روش کارخانه برای ایجاد کلاسی که یک عملیات EnqueueTPUEmbeddingBatch جدید را بسته بندی می کند. |
استاتیک EnqueueTPUEmbeddingBatch.Options | deviceOrdinal (Long deviceOrdinal) |
روش های ارثی
روش های عمومی
ترکیب کننده های عمومی استاتیک EnqueueTPUEmbeddingBatch.Options (ترکیب کننده های List<String>)
مولفه های
ترکیب کننده ها | فهرستی از اسکالرهای رشته ای، یکی برای هر جدول جاسازی که نحوه عادی سازی فعال سازی های جاسازی را پس از جمع وزنی مشخص می کند. ترکیب کننده های پشتیبانی شده 'mean'، 'sum' یا 'sqrtn' هستند. اینکه مجموع اوزان برای «میانگین» 0 باشد یا مجموع اوزان مجذور برای «sqrtn» 0 باشد، نامعتبر است. اگر ترکیبکنندهها تصویب نشد، پیشفرض استفاده از «جمع» برای همه جداول است. |
---|
عمومی ایستا EnqueueTPUEmbeddingBatch ایجاد ( دامنه دامنه ، تکرارپذیر< Operand <String>> دسته، عملوند <String> modeOverride، Options... گزینه ها)
روش کارخانه برای ایجاد کلاسی که یک عملیات EnqueueTPUEmbeddingBatch جدید را بسته بندی می کند.
مولفه های
محدوده | محدوده فعلی |
---|---|
دسته ای | فهرستی از تانسورهای 1 بعدی، یکی برای هر جدول جاسازی، حاوی ورودیهای دستهای که بهعنوان پروتوهای dist_belief.SparseFeatures کدگذاری شدهاند. اگر فیلد وزن در پروتو SparseFeatures برای شناسه پر نشده باشد، وزن 1.0 در نظر گرفته می شود. |
modeOverride | ورودی رشته ای که حالت مشخص شده در TPUEmbeddingConfiguration را لغو می کند. مقادیر پشتیبانی شده عبارتند از: {'نامشخص'، 'استنتاج'، 'training'، 'backward_pass_only'}. وقتی روی "نا مشخص" تنظیم شود، حالت تنظیم شده در TPUEmbeddingConfiguration استفاده می شود، در غیر این صورت mode_override استفاده می شود. |
گزینه ها | مقادیر ویژگی های اختیاری را حمل می کند |
برمی گرداند
- یک نمونه جدید از EnqueueTPUEmbeddingBatch
عمومی ثابت EnqueueTPUEmbeddingBatch.Options deviceOrdinal (طولانی دستگاهOrdinal)
مولفه های
دستگاه عادی | دستگاه TPU برای استفاده. زمانی که Op در دستگاه TPU در حال اجرا است، این مقدار باید -1 باشد، و زمانی که Op روی دستگاه CPU اجرا می شود >= 0 باشد. |
---|