XlaSplitND

מחלקה גמר ציבורית XlaSplitND

מפצל טנזור קלט על פני כל הממדים.

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

אופציה זו עשויה להיווצר באמצעות גשר ה-TPU.

לדוגמה, עם טנסור `קלט`:

[[0, 1, 2],
  [3, 4, 5],
  [6, 7, 8]]
 

`num_splits`:

[2, 2]
 

ו-`paddings`:

[1, 1]
 

ה`פלטים` הצפויים הם:

[[0, 1],
  [3, 4]]
 [[2, 0],
  [5, 0]]
 [[6, 7],
  [0, 0]]
 [[8, 0],
  [0, 0]]
 

כיתות מקוננות

מעמד

XlaSplitND.Options

תכונות אופציונליות עבור XlaSplitND

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

סטטי <T> XlaSplitND <T>	ליצור ( היקף היקף, קלט Operand <T>, Long N, List<Long> numSplits, Options... options) שיטת מפעל ליצירת מחלקה העוטפת פעולת XlaSplitND חדשה.
איטרטור< Operand <T>>	איטרטור ()
רשימה< פלט <T>>	פלטים ()
סטטי XlaSplitND.Options	ריפודים (רפידות רשימה<Long>)

שיטות בירושה

מכיתה org.tensorflow.op.PrimitiveOp

בוליאנית סופית	שווה (Object obj)
int סופי	hashcode ()
מבצע	אופ () מחזירה את Operation הבסיסית
מחרוזת סופית	toString ()

מכיתה java.lang.Object

בוליאני	שווה (Object arg0)
שיעור אחרון<?>	getClass ()
int	hashcode ()
ריק סופי	להודיע ​​()
ריק סופי	הודע הכל ()
חוּט	toString ()
ריק סופי	המתן (ארג0 ארוך, int arg1)
ריק סופי	חכה (ארג0 ארוך)
ריק סופי	חכה ()

מהממשק java.lang.Iterable

בָּטֵל	forEach (Consumer<? super T> arg0)
איטרטור מופשט< Oprand <T>>	איטרטור ()
מפצל < Oprand <T>>	מפצל ()

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

סטטי ציבורי XlaSplitND <T> ליצור ( היקף היקף , קלט Operand <T>, Long N, List<Long> numSplits, Options... אפשרויות)

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

פרמטרים

תְחוּם	ההיקף הנוכחי
קֶלֶט	קלט טנסור לפיצול על פני כל הממדים. } out_arg { name: "outputs" description: <
numSplits	מספר דרכים לפיצול לכל מימד. מידות הצורה חייבות להיות ניתנות לחלוקה שווה.
אפשרויות	נושא ערכי תכונות אופציונליות

החזרות

• מופע חדש של XlaSplitND

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

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

רפידות XlaSplitND.Options סטטיות ציבוריות (רפידות רשימה<Long>)

פרמטרים

ריפודים	רשימה אופציונלית של ריפודים ימניים לכל מימד של טנזור קלט ליישום לפני הפיצול. זה יכול לשמש כדי להפוך ממד לחלוקה שווה.