TPUPartitionedOutputV2

מחלקה סופית ציבורית TPUPartitionedOutputV2

אופציה שמפיצה טנזור לריסוק על ידי XLA לרשימה של מחיצות

פלטים מחוץ לחישוב XLA. תומך ברסיס ND.

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

סטטי <T> TPUPartitionedOutputV2 <T>
ליצור ( היקף היקף , כניסות Operand <T>, Long numSplits, List<Long> partitionDims)
שיטת מפעל ליצירת מחלקה העוטפת פעולת TPUPartitionedOutputV2 חדשה.
איטרטור< Operand <T>>
רשימה< פלט <T>>
פלט ()
רשימה של פלטים מחולקים בעלי אותה צורה.

שיטות בירושה

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

public static TPUPartitionedOutputV2 <T> create ( scope scope, כניסות Operand <T>, Long numSplits, List<Long> partitionDims)

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

פרמטרים
תְחוּם ההיקף הנוכחי
תשומות טנזור המייצג את הצורה המלאה של טנסורים מחולקים.
partitionDims רשימה של מספרים שלמים המתארים כיצד כל מימד מחולק. ריקנות מציינת שהכניסות משוכפלות.
החזרות
  • מופע חדש של TPUPartitionedOutputV2

איטרטור ציבורי< Operand <T>> איטרטור ()

רשימה ציבורית< פלט <T>> פלט ()

רשימה של פלטים מחולקים בעלי אותה צורה.