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

NcclReduce

המעמד הסופי הציבור NcclReduce

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

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

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

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

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

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

שיטות בירושה

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

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

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

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

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

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

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

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