XlaRecvTPUEmbeddingActivations

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

عملیاتی که فعال‌سازی‌های تعبیه‌شده را روی TPU دریافت می‌کند.

سیستم TPU جستجوهای جاسازی و تجمیع را انجام می دهد. نتایج این تجمیع‌ها برای نمودار Tensorflow به‌عنوان خروجی‌های یک XlaRecvTPUEmbeddingActivations عملیات قابل مشاهده است. این عملیات فهرستی حاوی یک تانسور فعال سازی در هر جدول مشخص شده در مدل را برمی گرداند.

روش های عمومی

استاتیک XlaRecvTPUEmbeddingActivations
ایجاد ( scope scope، Operand <?> deduplicationData، Long numTables، String config)
روش کارخانه برای ایجاد کلاسی که یک عملیات XlaRecvTPUEmbeddingActivations جدید را بسته بندی می کند.
Iterator< عملوند <Float>>
فهرست< خروجی <شناور>>
خروجی ها ()
TensorList از فعال‌سازی‌های تعبیه‌شده حاوی یک تانسور در هر جدول جاسازی در مدل.

روش های ارثی

روش های عمومی

استاتیک عمومی XlaRecvTPUEmbeddingActivations ایجاد ( حوزه دامنه ، عملوند <?> deduplicationData، NumTables طولانی، پیکربندی رشته)

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

مولفه های
محدوده محدوده فعلی
DeduplicationData یک تانسور با نوع=DT_VARIANT که حاوی داده‌های تکراری است. تانسور یک تاپل تو در تو XLA حاوی N عنصر است (که در آن N نسبت تعداد هسته های جاسازی شده به هسته های تانسور در هر تراشه TPU است). هر عنصر از تاپل تودرتو یک تاپل از تانسورهای رتبه 1 است. هر تانسور یا شامل شاخص‌هایی (DT_UINT32) برای جاسازی جستجو در TensorCore یا وزن‌هایی (DT_FLOAT) برای اعمال در خروجی عملیات جستجوی جاسازی است.
numTables تعداد تانسورهای فعال سازی خروجی. اگر توصیفگر ویژگی در پیکربندی جاسازی tpu وجود داشته باشد، برابر با تعداد ویژگی‌ها در غیر این صورت برابر با تعداد جداول جاسازی در مدل است.
پیکربندی پروتوی TPUEmbeddingConfiguration سریال.
برمی گرداند
  • یک نمونه جدید از XlaRecvTPUEmbeddingActivations

عمومی Iterator< عملوند <Float>> تکرار کننده ()

لیست عمومی< خروجی <شناور>> خروجی ()

TensorList از فعال‌سازی‌های تعبیه‌شده حاوی یک تانسور در هر جدول جاسازی در مدل.