מוציא טנזור המכיל את ההפחתה על פני כל טנסור הקלט.
מוציא טנסור המכיל את ההפחתה על פני כל טנסור הקלט שהועברו ל-ops בתוך אותו `shared_name.
הגרף צריך להיות בנוי כך שאם הפעלה אחת פועלת עם ערך shared_name `c`, אז פעולות של `num_devices` יפעלו עם ערך shared_name `c`. אי ביצוע פעולה זו יגרום לכשל בביצוע הגרף.
קלט: הקלט לנתוני ההפחתה: ערך ההפחתה בכל מכשירי `num_devices`. צמצום: פעולת ההפחתה שיש לבצע. num_devices: מספר המכשירים המשתתפים בהפחתה זו. shared_name: מזהה ששיתף בין פעולות של אותה הפחתה.
שיטות ציבוריות
פלט <T> | asOutput () מחזירה את הידית הסמלית של טנזור. |
סטטי <T מרחיב מספר> NcclAllReduce <T> | |
פלט <T> | נתונים () |
שיטות בירושה
שיטות ציבוריות
פלט ציבורי <T> asOutput ()
מחזירה את הידית הסמלית של טנזור.
כניסות לפעולות TensorFlow הן יציאות של פעולת TensorFlow אחרת. שיטה זו משמשת להשגת ידית סמלית המייצגת את חישוב הקלט.
public static NcclAllReduce <T> create ( היקף היקף , קלט Operand <T>, הפחתת מחרוזת, מספר מכשירים ארוכים, מחרוזת משותף שם)
שיטת מפעל ליצירת מחלקה העוטפת פעולת NcclAllReduce חדשה.
פרמטרים
תְחוּם | ההיקף הנוכחי |
---|
החזרות
- מופע חדש של NcclAllReduce