عملیاتی که فعالسازیهای تعبیهشده را روی TPU دریافت میکند.
سیستم TPU جستجوها و تجمیعهای تعبیهشده را انجام میدهد که توسط آرگومانهای TPUEmbeddingEnqueue(Integer/Sparse/SparseTensor)Batch مشخص شدهاند. نتایج این تجمیعها برای نمودار Tensorflow به عنوان خروجیهای یک عملیات RecvTPUEmbeddingActivations قابل مشاهده است. این عملیات فهرستی حاوی یک تانسور فعال سازی در هر جدول مشخص شده در مدل را برمی گرداند. حداکثر یک عملیات RecvTPUEmbeddingActivations در نمودار TPU وجود دارد.
ثابت ها
رشته | OP_NAME | نام این عملیات، همانطور که توسط موتور هسته TensorFlow شناخته می شود |
روش های عمومی
استاتیک RecvTPUEmbeddingActivations | |
Iterator< Operand < TFloat32 >> | اشاره گر () |
لیست< خروجی < TFloat32 >> | خروجی ها () TensorList از فعالسازیهای تعبیهشده حاوی یک تانسور در هر جدول جاسازی در مدل. |
روش های ارثی
ثابت ها
رشته نهایی ثابت عمومی OP_NAME
نام این عملیات، همانطور که توسط موتور هسته TensorFlow شناخته می شود
روش های عمومی
ایجاد RecvTPUEmbeddingActivations استاتیک عمومی ( حوزه دامنه ، تعداد خروجی های طولانی، پیکربندی رشته)
روش کارخانه برای ایجاد کلاسی که یک عملیات RecvTPUEmbeddingActivations جدید را بسته بندی می کند.
مولفه های
محدوده | محدوده فعلی |
---|---|
numOutputs | تعداد تانسورهای فعال سازی خروجی، برابر با تعداد جداول تعبیه شده در مدل. |
پیکربندی | پروتوی TPUEmbeddingConfiguration سریال. |
برمی گرداند
- یک نمونه جدید از RecvTPUEmbeddingActivations