לוקח את המספר הנתון של אלמנטים שהושלמו ממחסום.
פעולה זו משרשרת טנסורים של רכיבים שלמים לאורך הממד ה-0 כדי ליצור טנסור רכיב בודד.
אלמנטים יוצאים מהמחסום כשהם שלמים, ובסדר שבו הוכנסו למחסום. פלט המדדים מספק מידע על האצווה שבה כל אלמנט הוכנס במקור למחסום.
כיתות מקוננות
מעמד | BarrierTakeMany.Options | תכונות אופציונליות עבור BarrierTakeMany |
שיטות ציבוריות
סטטי BarrierTakeMany.Options | allowSmallBatch (בוליאני allowSmallBatch) |
BarrierTakeMany סטטי | create ( scope scope, Operand <String> handle, Operand <Integer> numElements, List<Class<?>> componentTypes, Options... options) שיטת מפעל ליצירת מחלקה העוטפת פעולת BarrierTakeMany חדשה. |
פלט <Long> | מדדים () טנזור חד מימדי של מדדים, עם אורך num_elems. |
פלט <מחרוזת> | מפתחות () טנסור חד מימדי של מפתחות, עם אורך num_elements. |
סטטי BarrierTakeMany.Options | timeoutMs (Long timeoutMs) |
רשימה< פלט <?>> | ערכים () טנזור אחד בכל מימד לכל רכיב באלמנט מחסום. |
סטטי BarrierTakeMany.Options | waitForIncomplete (בוליאני waitForIncomplete) |
שיטות בירושה
שיטות ציבוריות
Public static BarrierTakeMany.Options allowSmallBatch (Boolian allowSmallBatch)
פרמטרים
allowSmallBatch | אפשר להחזיר פחות מ-num_elements פריטים אם המחסום כבר סגור. |
---|
public static BarrierTakeMany create ( scope scope, Operand <String> handle, Operand <Integer> numElements, List<Class<?>> componentTypes, Options... options)
שיטת מפעל ליצירת מחלקה העוטפת פעולת BarrierTakeMany חדשה.
פרמטרים
תְחוּם | ההיקף הנוכחי |
---|---|
ידית | הידית למחסום. |
numElements | טנזור בעל אלמנט בודד המכיל את מספר האלמנטים שיש לקחת. |
רכיבים | הסוג של כל רכיב בערך. |
אפשרויות | נושא ערכי תכונות אופציונליות |
החזרות
- מופע חדש של BarrierTakeMany
מדדי פלט ציבוריים <Long> ()
טנזור חד מימדי של מדדים, עם אורך num_elems. מדדים אלו מתייחסים לאצווה שבה הוצבו הערכים במחסום (החל מ-MIN_LONG ועולה עם כל BarrierInsertMany).
public static BarrierTakeMany.Options timeoutMs (Long timeoutMs)
פרמטרים
timeoutMs | אם התור ריק, פעולה זו תיחסם לעד timeout_ms אלפיות השנייה. הערה: אפשרות זו אינה נתמכת עדיין. |
---|
ערכי רשימה ציבורית< פלט <?>> ()
טנזור אחד בכל מימד לכל רכיב באלמנט מחסום. לכל הערכים יש אורך num_elements בממד ה-0.