EnqueueTPUEmbeddingSparseTensorBatch

מחלקה סופית ציבורית EnqueueTPUEmbeddingSparseTensorBatch

מקל על העברה של קוד המשתמש ב-tf.nn.embedding_lookup_sparse().

sample_index[i], embedding_index[i] ומשקל_צבירה[i] תואמים לתכונה ith. table_ids[i] מציין איזו טבלת הטמעה לחפש את התכונה.

הטנזורים במיקומים תואמים בשלושת רשימות הקלט (מדדי_מדגם, מדדי_הטבעה ומשקולות_צבירה) חייבים להיות בעלי אותה צורה, כלומר דרגה 1 עם dim_size() שווה למספר הכולל של חיפושים בטבלה המתוארת על ידי התכונה המתאימה.

כיתות מקוננות

מעמד EnqueueTPUEbeddingSparseTensorBatch.Options תכונות אופציונליות עבור EnqueueTPUEmbeddingSparseTensorBatch

שיטות ציבוריות

סטטי EnqueueTPUEmbeddingSparseTensorBatch.Options
שילובי (List<String> שילובים)
סטטי <T מרחיב את המספר, U מרחיב את המספר, V מרחיב את המספר> EnqueueTPUEmbeddingSparseTensorBatch
create ( scope scope, Iterable< Operand <T>> sampleIndices, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand <String> modeOverride, List<Long> tableIds, Options... options)
שיטת מפעל ליצירת מחלקה העוטפת פעולת EnqueueTPUEmbeddingSparseTensorBatch חדשה.
סטטי EnqueueTPUEmbeddingSparseTensorBatch.Options
deviceOrdinal (Long deviceOrdinal)
סטטי EnqueueTPUEmbeddingSparseTensorBatch.Options
maxSequenceLengths (רשימה<Long> maxSequenceLengths)
סטטי EnqueueTPUEmbeddingSparseTensorBatch.Options
numFeatures (רשימה<Long> numFeatures)

שיטות בירושה

שיטות ציבוריות

public static EnqueueTPUEmbeddingSparseTensorBatch.Options combiners (List<String> combiners)

פרמטרים
משלבים רשימה של סקלרים של מחרוזת, אחת לכל טבלת הטמעה המציינת כיצד לנרמל את הפעלות ההטבעה לאחר סיכום משוקלל. המשלבים הנתמכים הם 'mean', 'sum' או 'sqrtn'. זה לא חוקי שסכום המשקולות יהיה 0 עבור 'ממוצע' או שסכום המשקולות בריבוע יהיה 0 עבור 'sqrtn'. אם המשלבים לא עוברים, ברירת המחדל היא להשתמש ב-'sum' עבור כל הטבלאות.

public static EnqueueTPUEmbeddingSparseTensorBatch create ( scope scope, Iterable< Operand <T>> sampleIndices, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand <String> modeOverride, List<Long> tablesId . . אפשרויות)

שיטת מפעל ליצירת מחלקה העוטפת פעולת EnqueueTPUEmbeddingSparseTensorBatch חדשה.

פרמטרים
תְחוּם ההיקף הנוכחי
מדדים לדוגמה רשימה של טנסורים בדרגה 1 המציינת את דוגמה לאימון שאליה שייכים ערכי ההטבעה והמשקלים המתאימים. זה מתאים ל-sp_ids.indices[:,0] ב-embedding_lookup_sparse().
הטבעה מדדים רשימה של טנסור דרגה 1, מדדים לטבלאות ההטמעה. זה מתאים ל-sp_ids.values ​​ב-embedding_lookup_sparse().
משקלי צבירה רשימה של טנסור דרגה 1 המכילה משקלי צבירה לדוגמה לכל אימון. זה מתאים ל-sp_weights.values ​​ב-embedding_lookup_sparse().
מצב ביטול קלט מחרוזת שעוקף את המצב שצוין ב-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.
אפשרויות נושא ערכי תכונות אופציונליות
החזרות
  • מופע חדש של EnqueueTPUEmbeddingSparseTensorBatch

public static EnqueueTPUEmbeddingSparseTensorBatch.Options deviceOrdinal (Long deviceOrdinal)

פרמטרים
deviceOrdinal מכשיר ה-TPU לשימוש. צריך להיות >= 0 וקטן ממספר ליבות ה-TPU במשימה שעליה ממוקם הצומת.

public static EnqueueTPUEmbeddingSparseTensorBatch.Options maxSequenceLengths (רשימה<Long> maxSequenceLengths)

public static EnqueueTPUEbeddingSparseTensorBatch.Options numFeatures (רשימה<Long> numFeatures)