מחלקה סופית ציבורית TPUPartitionedOutputV2
אופציה שמפיצה טנזור לריסוק על ידי XLA לרשימה של מחיצות
פלטים מחוץ לחישוב XLA. תומך ברסיס ND.
שיטות ציבוריות
סטטי <T> TPUPartitionedOutputV2 <T> | |
איטרטור< Operand <T>> | איטרטור () |
רשימה< פלט <T>> | פלט () רשימה של פלטים מחולקים בעלי אותה צורה. |
שיטות בירושה
שיטות ציבוריות
public static TPUPartitionedOutputV2 <T> create ( scope scope, כניסות Operand <T>, Long numSplits, List<Long> partitionDims)
שיטת מפעל ליצירת מחלקה העוטפת פעולת TPUPartitionedOutputV2 חדשה.
פרמטרים
תְחוּם | ההיקף הנוכחי |
---|---|
תשומות | טנזור המייצג את הצורה המלאה של טנסורים מחולקים. |
partitionDims | רשימה של מספרים שלמים המתארים כיצד כל מימד מחולק. ריקנות מציינת שהכניסות משוכפלות. |
החזרות
- מופע חדש של TPUPartitionedOutputV2