NcclAllReduce

כיתת גמר ציבורית NcclAllReduce

מוציא טנזור המכיל את ההפחתה על פני כל טנסור הקלט.

מוציא טנסור המכיל את ההפחתה על פני כל טנסור הקלט שהועברו ל-ops בתוך אותו `shared_name.

הגרף צריך להיות בנוי כך שאם הפעלה אחת פועלת עם ערך shared_name `c`, אז פעולות של `num_devices` יפעלו עם ערך shared_name `c`. אי ביצוע פעולה זו יגרום לכשל בביצוע הגרף.

קלט: הקלט לנתוני ההפחתה: ערך ההפחתה בכל מכשירי `num_devices`. צמצום: פעולת ההפחתה שיש לבצע. num_devices: מספר המכשירים המשתתפים בהפחתה זו. shared_name: מזהה ששיתף בין פעולות של אותה הפחתה.

קבועים

חוּט

OP_NAME

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

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

פלט <T>	asOutput () מחזירה את הידית הסמלית של הטנזור.
סטטי <T מרחיב את TNummer > NcclAllReduce <T>	ליצור (היקף היקף , קלט Operand <T>, הפחתת מחרוזת, מספר מכשירים ארוכים, שם משותף מחרוזת) שיטת מפעל ליצירת מחלקה העוטפת פעולת NcclAllReduce חדשה.
פלט <T>	נתונים ()

שיטות בירושה

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

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

מכיתה java.lang.Object

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

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

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

מממשק org.tensorflow.Operand

פלט מופשט <T>	asOutput () מחזירה את הידית הסמלית של הטנזור.
תקציר T	asTensor () מחזיר את הטנזור באופרנד זה.
צורה מופשטת	צורה () מחזירה את הצורה (הידועה אולי בחלקה) של הטנזור שאליו מתייחס Output של אופרנד זה.
מחלקה מופשטת<T>	סוג () מחזירה את סוג הטנזור של אופרנד זה

מהממשק org.tensorflow.ndarray.Shaped

מופשט int	דרגה ()
צורה מופשטת	צורה ()
מופשט ארוך	גודל () מחשב ומחזיר את הגודל הכולל של מיכל זה, במספר ערכים.

קבועים

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

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

ערך קבוע: "NcclAllReduce"

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

פלט ציבורי <T> asOutput ()

מחזירה את הידית הסמלית של הטנזור.

כניסות לפעולות TensorFlow הן יציאות של פעולת TensorFlow אחרת. שיטה זו משמשת להשגת ידית סמלית המייצגת את חישוב הקלט.

public static NcclAllReduce <T> create ( היקף היקף , קלט Operand <T>, הפחתת מחרוזת, מספר מכשירים ארוכים, מחרוזת משותף שם)

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

פרמטרים

תְחוּם	ההיקף הנוכחי

החזרות

• מופע חדש של NcclAllReduce

פלט ציבורי <T> נתוני ()