מחלקה סופית ציבורית EnqueueTPUEmbeddingBatch
הפעלה שמעמידה בתור רשימה של טנזורי אצווה קלט ל-TPUEmbedding.
הפעלה שמעמידה בתור רשימה של טנזורי אצווה קלט ל-TPUEmbedding.
כיתות מקוננות
מעמד | EnqueueTPUEmbeddingBatch.Options | תכונות אופציונליות עבור EnqueueTPUEmbeddingBatch |
שיטות ציבוריות
סטטי EnqueueTPUEmbeddingBatch.Options | שילובי (List<String> שילובים) |
סטטי EnqueueTPUEmbeddingBatch | ליצור ( היקף היקף, Iterable< Operand <String>> אצווה, Operand <String> modeOverride, Options... אפשרויות) שיטת מפעל ליצירת מחלקה העוטפת פעולת EnqueueTPUEmbeddingBatch חדשה. |
סטטי EnqueueTPUEmbeddingBatch.Options | deviceOrdinal (Long deviceOrdinal) |
שיטות בירושה
שיטות ציבוריות
public static EnqueueTPUEmbeddingBatch.Options combiners (List<String> combiners)
פרמטרים
משלבים | רשימה של סקלרים של מחרוזת, אחת לכל טבלת הטמעה המציינת כיצד לנרמל את הפעלות ההטבעה לאחר סיכום משוקלל. המשלבים הנתמכים הם 'mean', 'sum' או 'sqrtn'. זה לא חוקי שסכום המשקולות יהיה 0 עבור 'ממוצע' או שסכום המשקולות בריבוע יהיה 0 עבור 'sqrtn'. אם המשלבים לא עוברים, ברירת המחדל היא להשתמש ב-'sum' עבור כל הטבלאות. |
---|
public static EnqueueTPUEmbeddingBatch create ( היקף היקף , Iterable< Operand <String>> אצווה, Operand <String> modeOverride, Options... אפשרויות)
שיטת מפעל ליצירת מחלקה העוטפת פעולת EnqueueTPUEmbeddingBatch חדשה.
פרמטרים
תְחוּם | ההיקף הנוכחי |
---|---|
קבוצה | רשימה של טנזורים 1D, אחד עבור כל טבלת הטבעה, המכילה את כניסות האצווה המקודדות כפרטואים dist_belief.SparseFeatures. אם שדה המשקל בפרוטו SparseFeatures אינו מאוכלס עבור מזהה, יש להניח משקל של 1.0. |
מצב ביטול | קלט מחרוזת שעוקף את המצב שצוין ב-TPUEmbeddingConfiguration. הערכים הנתמכים הם {'unspecificed', 'inference', 'training', 'backward_pass_only'}. כאשר מוגדר ל'unspecificed', נעשה שימוש במצב שהוגדר ב-TPUEmbeddingConfiguration, אחרת נעשה שימוש ב-mode_override. |
אפשרויות | נושא ערכי תכונות אופציונליות |
החזרות
- מופע חדש של EnqueueTPUEmbeddingBatch
public static EnqueueTPUEmbeddingBatch.Options deviceOrdinal (Long deviceOrdinal)
פרמטרים
deviceOrdinal | מכשיר ה-TPU לשימוש. זה צריך להיות -1 כאשר ה-Op פועל על התקן TPU, ו->= 0 כאשר ה-Op פועל על התקן ה-CPU. |
---|