NcclAllReduce

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

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

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

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

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

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

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

שיטות בירושה

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

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

מכיתה java.lang.Object

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

מממשק org.tensorflow.Operand

פלט מופשט <T>

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

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

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

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

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

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

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

פרמטרים

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

החזרות

• מופע חדש של NcclAllReduce

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