מחלקה סטטית ציבורית EnqueueTPUEmbeddingBatch.Options
תכונות אופציונליות עבור EnqueueTPUEmbeddingBatch
שיטות ציבוריות
EnqueueTPUEmbeddingBatch.Options | שילובי (List<String> שילובים) |
EnqueueTPUEmbeddingBatch.Options | deviceOrdinal (Long deviceOrdinal) |
שיטות בירושה
שיטות ציבוריות
public EnqueueTPUEmbeddingBatch.Options combiners (List<String> combiners)
פרמטרים
משלבים | רשימה של סקלרים של מחרוזת, אחת לכל טבלת הטמעה המציינת כיצד לנרמל את הפעלות ההטבעה לאחר סיכום משוקלל. המשלבים הנתמכים הם 'mean', 'sum' או 'sqrtn'. זה לא חוקי שסכום המשקולות יהיה 0 עבור 'ממוצע' או שסכום המשקולות בריבוע יהיה 0 עבור 'sqrtn'. אם המשלבים לא עוברים, ברירת המחדל היא להשתמש ב-'sum' עבור כל הטבלאות. |
---|
public EnqueueTPUEmbeddingBatch.Options deviceOrdinal (Long deviceOrdinal)
פרמטרים
deviceOrdinal | מכשיר ה-TPU לשימוש. זה צריך להיות -1 כאשר ה-Op פועל על התקן TPU, ו->= 0 כאשר ה-Op פועל על התקן ה-CPU. |
---|