NcclReduce

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

מפחית 'קלט' מ-'num_devices' באמצעות 'reduction' למכשיר בודד.

מפחית 'קלט' מ-'num_devices' באמצעות 'reduction' למכשיר בודד.

הגרף צריך להיות בנוי כך שלכל התשומות תהיה הקצאת התקן חוקית, ולאופ עצמו מוקצה אחד מההתקנים הללו.

קלט: הקלט להפחתה. נתונים: ערך ההפחתה בכל מכשירי `num_devices`. צמצום: פעולת ההפחתה שיש לבצע.

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

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

שיטות בירושה

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

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

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

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

public static NcclReduce <T> create ( היקף היקף , קלט איטרני< Oprand <T>>, הפחתת מחרוזת)

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

פרמטרים
תְחוּם ההיקף הנוכחי
החזרות
  • מופע חדש של NcclReduce

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