EnqueueTPUEmbeddingBatch

מחלקה סופית ציבורית 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.