מקבץ את כל טנסור הקלט בצורה לא דטרמיניסטית.
כאשר מופעים רבים של ה-Op הזה מופעלים במקביל עם אותו מיכל/שם_משותף באותו התקן, חלקם יוציאו Tensors בצורת אפס ואחרים יציגו Tensors בגודל של עד max_batch_size.
כל הטנסורים ב-in_tensors מצטברים יחד (לכן, למשל, תוויות ותכונות צריכות להיות אצווה עם מופע בודד של פעולה זו.
כל הפעלת אצווה פולטת סקלאר 'id' אשר ישמש כדי לזהות הפעלה מסוימת זו בעת ביצוע ביטול אצווה או הדרגתיות שלו.
כל הפעלה שפולטת אצווה לא ריקה תפלוט גם אצווה לא ריקה Tensor, שהיא מטריצה [K, 3] שבה כל שורה מכילה את המזהה, ההתחלה והאורך של הרכיבים של כל קבוצה של Tensors ב- אצווה_טנסורים.
טנזורים קבוצתיים משורשרים לאורך הממד הראשון, וכל הטנזורים ב-in_tensורים חייבים להיות בעלי הממד הראשון באותו גודל.
in_tensors: הטנזורים שיש לקבץ. num_batch_threads: מספר שרשורי תזמון לעיבוד קבוצות עבודה. קובע את מספר האצוות המעובדות במקביל. max_batch_size: גדלי אצווה לעולם לא יהיו גדולים מזה. batch_timeout_micros: מספר מקסימלי של מיקרו-שניות להמתנה לפני הפלט של אצווה לא שלמה. allow_batch_sizes: רשימה אופציונלית של גדלי אצווה מותרים. אם נשאר ריק, לא עושה כלום. אחרת, מספק רשימה של גדלי אצווה, מה שגורם לאופ לרפד אצווה עד לאחד מהגדלים הללו. הערכים חייבים לגדול באופן מונוטוני, והערך הסופי חייב להיות שווה ל-max_batch_size. grad_timeout_micros: הזמן הקצוב לשימוש עבור השיפוע. ראה ביטול אצווה. batched_tensors: או טנסורים ריקים או אצווה של טנסורים משורשרים. batch_index: אם out_tensors אינו ריק, יש מידע כדי להפוך אותו. מיכל: שולט בהיקף השיתוף של אצווה זו. id: מכיל תמיד סקלאר עם מזהה ייחודי להפעלת אצווה זו. shared_name: פועלים בו-זמנית של מופעים של אצווה באותו מכשיר עם אותו מיכל ו-shared_name יציגו את הרכיבים שלהם יחד. אם יישאר ריק, שם הפעולה ישמש כשם המשותף. T: סוגי הטנסורים שיש לקבץ.
כיתות מקוננות
| מעמד | אצווה.אפשרויות | תכונות אופציונליות עבור Batch | |
שיטות ציבוריות
| סטטי Batch.Options | allowBatchSizes (רשימה<Long> allowBatchSizes) |
| פלט <Long> | batchIndex () |
| רשימה< פלט <?>> | |
| סטטי Batch.Options | batchingQueue (מחרוזת batchingQueue) |
| סטטי Batch.Options | מיכל (מיכל מחרוזת) |
| אצווה סטטית | צור ( scope scope, Iterable< Operand <?>> inTensors, Long numBatchThreads, Long maxBatchSize, Long batchTimeoutMicros, Long gradTimeoutMicros, Options... options) שיטת מפעל ליצירת מחלקה העוטפת פעולת אצווה חדשה. |
| פלט <Long> | מזהה () |
| סטטי Batch.Options | maxEnqueuedBatches (Long maxEnqueuedBatches) |
| סטטי Batch.Options | sharedName (מחרוזת sharedName) |
שיטות בירושה
שיטות ציבוריות
יצירת אצווה ציבורית ( scope scope, Iterable< Operand <?>> inTensors, Long numBatchThreads, Long maxBatchSize, Long batchTimeoutMicros, Long gradTimeoutMicros, Options... אפשרויות)
שיטת מפעל ליצירת מחלקה העוטפת פעולת אצווה חדשה.
פרמטרים
| תְחוּם | ההיקף הנוכחי |
|---|---|
| אפשרויות | נושא ערכי תכונות אופציונליות |
החזרות
- מופע חדש של Batch