מחלקה סופית ציבורית EnqueueTPUEmbeddingIntegerBatch
הפעלה שמעמידה בתור רשימה של טנזורי אצווה קלט ל-TPUEmbedding.
כיתות מקוננות
| מעמד | EnqueueTPUEmbeddingIntegerBatch.Options | תכונות אופציונליות עבור EnqueueTPUEmbeddingIntegerBatch | |
שיטות ציבוריות
| סטטי EnqueueTPUEmbeddingIntegerBatch | צור ( היקף היקף, Iterable< Operand <Integer>> אצווה, Operand <String> modeOverride, Options... אפשרויות) שיטת מפעל ליצירת מחלקה העוטפת פעולת EnqueueTPUEmbeddingIntegerBatch חדשה. |
| סטטי EnqueueTPUEmbeddingIntegerBatch.Options | deviceOrdinal (Long deviceOrdinal) |
שיטות בירושה
שיטות ציבוריות
public static EnqueueTPUEmbeddingIntegerBatch create ( scope scope, Iterable< Operand <Integer>> batch, Operand <String> modeOverride, Options... options)
שיטת מפעל ליצירת מחלקה העוטפת פעולת EnqueueTPUEmbeddingIntegerBatch חדשה.
פרמטרים
| תְחוּם | ההיקף הנוכחי |
|---|---|
| קבוצה | רשימה של טנזורים 1D, אחד לכל טבלת הטבעה, המכילה את המדדים לטבלאות. |
| מצב ביטול | קלט מחרוזת שעוקף את המצב שצוין ב-TPUEmbeddingConfiguration. הערכים הנתמכים הם {'unspecificed', 'inference', 'training', 'backward_pass_only'}. כאשר מוגדר ל'unspecificed', נעשה שימוש במצב שהוגדר ב-TPUEmbeddingConfiguration, אחרת נעשה שימוש ב-mode_override. |
| אפשרויות | נושא ערכי תכונות אופציונליות |
החזרות
- מופע חדש של EnqueueTPUEmbeddingIntegerBatch
public static EnqueueTPUEmbeddingIntegerBatch.Options deviceOrdinal (Long deviceOrdinal)
פרמטרים
| deviceOrdinal | מכשיר ה-TPU לשימוש. צריך להיות >= 0 וקטן ממספר ליבות ה-TPU במשימה שעליה ממוקם הצומת. |
|---|