מחלקה סופית ציבורית EnqueueTPUEmbeddingRaggedTensorBatch
מקל על העברה של קוד המשתמש ב-tf.nn.embedding_lookup().
sample_splits[i], embedding_index[i] ומשקולות_aggregation[i] מתאימות לתכונה ith. table_ids[i] מציין איזו טבלת הטמעה לחפש את התכונה.
הטנזורים במיקומים תואמים בשניים מרשימות הקלט, embedding_indices ו-aggregation_weights, חייבים להיות בעלי אותה צורה, כלומר דירוג 1 עם dim_size() שווה למספר הכולל של חיפושים בטבלה המתוארת על ידי התכונה המתאימה.
כיתות מקוננות
| מעמד | EnqueueTPUEmbeddingRaggedTensorBatch.Options | תכונות אופציונליות עבור EnqueueTPUEmbeddingRaggedTensorBatch | |
שיטות ציבוריות
| static EnqueueTPUEmbeddingRaggedTensorBatch.Options | שילובי (List<String> שילובים) |
| סטטי <T מרחיב את המספר, U מרחיב את המספר, V מרחיב את המספר> EnqueueTPUEmbeddingRaggedTensorBatch | |
| static EnqueueTPUEmbeddingRaggedTensorBatch.Options | deviceOrdinal (Long deviceOrdinal) |
| static EnqueueTPUEmbeddingRaggedTensorBatch.Options | maxSequenceLengths (רשימה<Long> maxSequenceLengths) |
| static EnqueueTPUEmbeddingRaggedTensorBatch.Options | numFeatures (רשימה<Long> numFeatures) |
שיטות בירושה
שיטות ציבוריות
public static EnqueueTPUEmbeddingRaggedTensorBatch.Options combiners (List<String> combiners)
פרמטרים
| משלבים | רשימה של סקלרים של מחרוזת, אחת לכל טבלת הטמעה המציינת כיצד לנרמל את הפעלות ההטבעה לאחר סיכום משוקלל. המשלבים הנתמכים הם 'mean', 'sum' או 'sqrtn'. זה לא חוקי שסכום המשקולות יהיה 0 עבור 'ממוצע' או שסכום המשקולות בריבוע יהיה 0 עבור 'sqrtn'. אם המשלבים לא עוברים, ברירת המחדל היא להשתמש ב-'sum' עבור כל הטבלאות. |
|---|
public static EnqueueTPUEmbeddingRaggedTensorBatch create ( scope scope, Iterable< Operand <T>> sampleSplits, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand <String> modeOverride, List<Long> tablesId . . אפשרויות)
שיטת מפעל ליצירת מחלקה העוטפת פעולת EnqueueTPUEmbeddingRaggedTensorBatch חדשה.
פרמטרים
| תְחוּם | ההיקף הנוכחי |
|---|---|
| מפוצלים לדוגמה | רשימה של טנסורים בדרגה 1 המציינת את נקודות השבירה לפיצול מדדי הטבעה ומשקולות צבירה לשורות. זה מתאים ל-ids.row_splits ב-embedding_lookup(), כאשר ids הוא RaggedTensor. |
| הטבעה מדדים | רשימה של טנסור דרגה 1, מדדים לטבלאות ההטמעה. זה מתאים ל-ids.values ב-embedding_lookup(), כאשר ids הוא RaggedTensor. |
| משקלי צבירה | רשימה של טנסור דרגה 1 המכילה משקלי צבירה לדוגמה לכל אימון. זה מתאים לשדה הערכים של RaggedTensor עם אותם row_splits כמו ids ב-embedding_lookup(), כאשר ids הוא RaggedTensor. |
| מצב ביטול | קלט מחרוזת שעוקף את המצב שצוין ב-TPUEmbeddingConfiguration. הערכים הנתמכים הם {'unspecificed', 'inference', 'training', 'backward_pass_only'}. כאשר מוגדר ל'unspecificed', נעשה שימוש במצב שהוגדר ב-TPUEmbeddingConfiguration, אחרת נעשה שימוש ב-mode_override. |
| מזהי טבלאות | רשימה של מספרים שלמים המציינת את המזהה של טבלת ההטמעה (היסט של TableDescriptor ב-TPUEmbeddingConfiguration) כדי לחפש את הקלט המתאים. הקלט ה-ith נבדק באמצעות table_ids[i]. הגודל של רשימת table_ids חייב להיות שווה לזה של sample_index, embedding_index ו-aggregation_weights. |
| אפשרויות | נושא ערכי תכונות אופציונליות |
החזרות
- מופע חדש של EnqueueTPUEmbeddingRaggedTensorBatch
public static EnqueueTPUEmbeddingRaggedTensorBatch.Options deviceOrdinal (Long deviceOrdinal)
פרמטרים
| deviceOrdinal | מכשיר ה-TPU לשימוש. צריך להיות >= 0 וקטן ממספר ליבות ה-TPU במשימה שעליה ממוקם הצומת. |
|---|