عملیاتی که پارامترهای بهینه سازی را از جاسازی در حافظه میزبان بازیابی می کند.
عملیاتی که پارامترهای بهینه سازی را از جاسازی در حافظه میزبان بازیابی می کند. باید قبل از آن یک عملیات ConfigureTPUEmbeddingHost باشد که پیکربندی صحیح جدول جاسازی را تنظیم می کند. به عنوان مثال، این عملیات برای بازیابی پارامترهای به روز شده قبل از ذخیره یک چک پوینت استفاده می شود. برای Adagrad، auxiliary1 پس از اجرای این عملیات حاوی انباشته‌کننده‌ها خواهد بود. برای SGD، تمام مقادیر کمکی* خالی خواهند بود (تانسورهای 0x0 برای آن جدول). برای FTRL، auxiliary1 حاوی انباشتگرها و auxiliary2 شامل عبارت های خطی است. برای ADAM، auxiliary1 حاوی لحظه و auxiliary2 حاوی سرعت خواهد بود.
روش های عمومی
| فهرست< خروجی <شناور>> | کمکی 1 () فهرستی از تانسورها، یکی برای هر جدول جاسازی، که حاوی اولین پارامتر بهینه سازی کمکی ذخیره شده است. |
| فهرست< خروجی <شناور>> | کمکی 2 () فهرستی از تانسورها، یکی برای هر جدول جاسازی، که حاوی دومین پارامتر بهینه سازی کمکی ذخیره شده است. |
| فهرست< خروجی <شناور>> | کمکی 3 () فهرستی از تانسورها، یکی برای هر جدول جاسازی، حاوی سومین پارامتر بهینه سازی کمکی ذخیره شده. |
| فهرست< خروجی <شناور>> | کمکی4 () فهرستی از تانسورها، یکی برای هر جدول جاسازی، که حاوی چهارمین پارامتر بهینه سازی کمکی ذخیره شده است. |
| فهرست< خروجی <شناور>> | کمکی 5 () فهرستی از تانسورها، یکی برای هر جدول جاسازی، که حاوی پنجمین پارامتر بهینه سازی کمکی ذخیره شده است. |
| فهرست< خروجی <شناور>> | کمکی 6 () فهرستی از تانسورها، یکی برای هر جدول جاسازی، که شامل شش پارامتر بهینه سازی کمکی ذخیره شده است. |
| فهرست< خروجی <شناور>> | کمکی 7 () فهرستی از تانسورها، یکی برای هر جدول جاسازی، که حاوی هفتمین پارامتر بهینه سازی کمکی ذخیره شده است. |
| Static RetrieveAllTPUEmbeddingParameters | |
| فهرست< خروجی <شناور>> | مولفه های () فهرستی از تانسورها، یکی برای هر جدول جاسازی، حاوی پارامترهای جدول جاسازی ذخیره شده. |
روش های ارثی
روش های عمومی
فهرست عمومی< خروجی <فلوت>> کمکی1 ()
فهرستی از تانسورها، یکی برای هر جدول جاسازی، که حاوی اولین پارامتر بهینه سازی کمکی ذخیره شده است. عناصر در لیست وجود دارند، اما اندازه صفر دارند، برای پارامترهای بهینه سازی استفاده نشده (بر اساس الگوریتم مورد استفاده برای هر جدول).
فهرست عمومی< خروجی <Float>> کمکی2 ()
فهرستی از تانسورها، یکی برای هر جدول جاسازی، که حاوی دومین پارامتر بهینه سازی کمکی ذخیره شده است. عناصر در لیست وجود دارند، اما اندازه صفر دارند، برای پارامترهای بهینه سازی استفاده نشده (بر اساس الگوریتم مورد استفاده برای هر جدول).
فهرست عمومی< خروجی <Float>> کمکی3 ()
فهرستی از تانسورها، یکی برای هر جدول جاسازی، حاوی سومین پارامتر بهینه سازی کمکی ذخیره شده. عناصر در لیست وجود دارند، اما اندازه صفر دارند، برای پارامترهای بهینه سازی استفاده نشده (بر اساس الگوریتم مورد استفاده برای هر جدول).
فهرست عمومی< خروجی <Float>> کمکی4 ()
فهرستی از تانسورها، یکی برای هر جدول جاسازی، که حاوی چهارمین پارامتر بهینه سازی کمکی ذخیره شده است. عناصر در لیست وجود دارند، اما اندازه صفر دارند، برای پارامترهای بهینه سازی استفاده نشده (بر اساس الگوریتم مورد استفاده برای هر جدول).
فهرست عمومی< خروجی <Float>> کمکی5 ()
فهرستی از تانسورها، یکی برای هر جدول جاسازی، که حاوی پنجمین پارامتر بهینه سازی کمکی ذخیره شده است. عناصر در لیست وجود دارند، اما اندازه صفر دارند، برای پارامترهای بهینه سازی استفاده نشده (بر اساس الگوریتم مورد استفاده برای هر جدول).
فهرست عمومی< خروجی <Float>> کمکی6 ()
فهرستی از تانسورها، یکی برای هر جدول جاسازی، که شامل شش پارامتر بهینه سازی کمکی ذخیره شده است. عناصر در لیست وجود دارند، اما اندازه صفر دارند، برای پارامترهای بهینه سازی استفاده نشده (بر اساس الگوریتم مورد استفاده برای هر جدول).
فهرست عمومی< خروجی <Float>> کمکی7 ()
فهرستی از تانسورها، یکی برای هر جدول جاسازی، که حاوی هفتمین پارامتر بهینه سازی کمکی ذخیره شده است. عناصر در لیست وجود دارند، اما اندازه صفر دارند، برای پارامترهای بهینه سازی استفاده نشده (بر اساس الگوریتم مورد استفاده برای هر جدول).
ایجاد RetrieveAllTPUEmbeddingParameters استاتیک عمومی ( حوزه دامنه، NumTables طولانی، پیکربندی رشته، numShards طولانی، Long shardId)
روش کارخانه برای ایجاد کلاسی که یک عملیات RetrieveAllTPUEmbeddingParameters جدید را بسته بندی می کند.
مولفه های
| محدوده | محدوده فعلی |
|---|---|
| NumTables | تعداد جداول تعبیه شده |
| پیکربندی | یک پروتوی TPUEmbeddingConfiguration که پارامترهای جدول در حال بارگذاری را توصیف می کند و به صورت سریال به رشته ای تبدیل می شود. |
| numShards | تعداد خرده هایی که جداول تعبیه شده به آنها تقسیم می شوند. |
| shardId | شناسه خرده برای این عملیات. |
برمی گرداند
- یک نمونه جدید از RetrieveAllTPUEmbeddingParameters
فهرست عمومی< خروجی <Float>> پارامترها ()
فهرستی از تانسورها، یکی برای هر جدول جاسازی، حاوی پارامترهای جدول جاسازی ذخیره شده.