אופ שמקבל הפעלות הטמעה ב-TPU.
מערכת ה-TPU מבצעת את חיפושי ההטמעה והצבירה שצוינו על ידי הארגומנטים ל-TPUEmbeddingEnqueue(Integer/Sparse/SparseTensor) Batch. התוצאות של צבירה אלו גלויות לגרף Tensorflow כפלטים של הפעלה של RecvTPUEembeddingActivations. פעולה זו מחזירה רשימה המכילה Tensor אחד של הפעלות לכל טבלה שצוינה במודל. יכולה להיות הפעלה אחת של RecvTPUEmbeddingActivations בתרשים ה-TPU לכל היותר.
שיטות ציבוריות
סטטי RecvTPUEmbeddingActivations | |
איטרטור< Oprand <Float>> | איטרטור () |
רשימה< פלט <צף>> | פלטים () רשימת Tensor של הפעלות הטמעה המכילה Tensor אחד לכל טבלת הטמעה במודל. |
שיטות בירושה
שיטות ציבוריות
Public static RecvTPUEmbeddingActivations create ( היקף היקף , Long numOutputs, String config)
שיטת מפעל ליצירת מחלקה העוטפת פעולת RecvTPUEmbeddingActivations חדשה.
פרמטרים
תְחוּם | ההיקף הנוכחי |
---|---|
numoutputs | מספר טנסור הפעלת הפלט, שווה למספר טבלאות הטבעה בדגם. |
תצורה | פרוטו TPUEmbeddingConfiguration מסודר. |
החזרות
- מופע חדש של RecvTPUEmbeddingActivations
רשימה ציבורית< פלט <צף>> פלטי ()
רשימת Tensor של הפעלות הטמעה המכילה Tensor אחד לכל טבלת הטמעה במודל.