EnqueueTPUEmbeddingSparseBatch

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

הפעלה שמעמידה בתור TPUEmbedding מדדי קלט מ-SparseTensor.

אופציה זו מקלה על העברה של קוד המשתמש ב-embedding_lookup_sparse(), אם כי עיבוד מקדים של Python של הארגומנטים SparseTensor ל-embedding_lookup_sparse() נדרש כדי לייצר את הארגומנטים ל-Op זה, מכיוון שרק אופ אחד של EnqueueTPUEmbeddingSparseBatch מותר לכל שלב אימון.

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

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

מעמד

EnqueueTPUEmbeddingSparseBatch.Options

תכונות אופציונליות עבור EnqueueTPUEmbeddingSparseBatch

קבועים

חוּט

OP_NAME

השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow

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

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

שיטות בירושה

מכיתה org.tensorflow.op.RawOp

בוליאנית סופית	שווה (Object obj)
int סופי	hashcode ()
מבצע	אופ () החזר יחידת חישוב זו Operation אחת.
מחרוזת סופית	toString ()

מכיתה java.lang.Object

בוליאני	שווה (Object arg0)
שיעור אחרון<?>	getClass ()
int	hashcode ()
ריק סופי	להודיע ​​()
ריק סופי	הודע הכל ()
חוּט	toString ()
ריק סופי	המתן (ארג0 ארוך, int arg1)
ריק סופי	חכה (ארג0 ארוך)
ריק סופי	חכה ()

מתוך ממשק org.tensorflow.op.Op

אבסטרקט ExecutionEnvironment	env () החזר את סביבת הביצוע שבה נוצר האופציה הזו.
מבצע מופשט	אופ () החזר יחידת חישוב זו Operation אחת.

קבועים

מחרוזת סופית סטטית ציבורית OP_NAME

השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow

ערך קבוע: "EnqueueTPUEmbeddingSparseBatch"

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

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

פרמטרים

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

public static EnqueueTPUEmbeddingSparseBatch create ( scope scope, Iterable< Operand <T>> sampleIndices, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand < TString > modeOverride, Options... options)

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

פרמטרים

תְחוּם	ההיקף הנוכחי
מדדים לדוגמה	רשימה של טנסורים בדרגה 1 המציינת את דוגמה והתכונה לאימון שאליהן שייכים ערכי ההטבעה והמשקלים המתאימים. sample_index[i] חייב להיות שווה ל-b *nf + f, כאשר nf הוא מספר התכונות מהטבלה המתאימה, f הוא ב-[0, nf), ו-b הוא ב-[0, גודל אצווה).
הטבעה מדדים	רשימה של טנסור דרגה 1, מדדים לטבלאות ההטמעה.
משקלי צבירה	רשימה של טנסורים בדרגה 1 המכילה לכל מדגם -- כלומר לכל (דוגמה לאימון, תכונה) -- משקלי צבירה.
מצב ביטול	קלט מחרוזת שעוקף את המצב שצוין ב-TPUEmbeddingConfiguration. הערכים הנתמכים הם {'unspecificed', 'inference', 'training', 'backward_pass_only'}. כאשר מוגדר ל'unspecificed', נעשה שימוש במצב שהוגדר ב-TPUEmbeddingConfiguration, אחרת נעשה שימוש ב-mode_override.
אפשרויות	נושא ערכי תכונות אופציונליות

החזרות

• מופע חדש של EnqueueTPUEmbeddingSparseBatch

public static EnqueueTPUEbeddingSparseBatch.Options deviceOrdinal (Long deviceOrdinal)

פרמטרים

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