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