TridiagonalSolve

שיעור הגמר הציבורי TridiagonalSolve

פותר מערכות משוואות תלת אלכסוניות.

פותר מערכות משוואות תלת אלכסוניות. תומך במידות אצווה ובמספר צדדים ימניים לכל צד שמאל. ב-CPU, הפתרון מחושב באמצעות חיסול גאוסי עם או בלי סיבוב חלקי, בהתאם לתכונה 'חלקית_ציר'. ב-GPU, נעשה שימוש בספריית cuSPARSE של Nvidia: https://docs.nvidia.com/cuda/cusparse/index.html#gtsv Pivoting חלקי עדיין לא נתמך על ידי קצה אחורי של XLA.

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

מעמד TridiagonalSolve.Options תכונות אופציונליות עבור TridiagonalSolve

קבועים

חוּט OP_NAME השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow

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

פלט <T>
asOutput ()
מחזירה את הידית הסמלית של הטנזור.
סטטי <T מרחיב את TType > TridiagonalSolve <T>
ליצור ( היקף היקף , אלכסוני Operand <T>, Operand <T> rhs, אפשרויות... אפשרויות)
שיטת מפעל ליצירת מחלקה העוטפת פעולת TridiagonalSolve חדשה.
פלט <T>
פלט ()
טנסור הצורה `[..., M, K]` המכיל את הפתרונות
סטטי TridiagonalSolve.Options
partialPivoting (בוליאני partialPivoting)

שיטות בירושה

org.tensorflow.op.RawOp
בוליאנית סופית
שווה (Object obj)
int סופי
hashcode ()
מבצע
אופ ()
החזר יחידת חישוב זו Operation אחת.
מחרוזת סופית
toString ()
בוליאני
שווה (Object arg0)
שיעור אחרון<?>
getClass ()
int
hashcode ()
ריק סופי
להודיע ​​()
ריק סופי
הודע הכל ()
חוּט
toString ()
ריק סופי
המתן (ארג0 ארוך, int arg1)
ריק סופי
המתן (ארג0 ארוך)
ריק סופי
חכה ()
org.tensorflow.op.Op
אבסטרקט ExecutionEnvironment
env ()
החזר את סביבת הביצוע שבה נוצר האופציה הזו.
מבצע מופשט
אופ ()
החזר יחידת חישוב זו Operation אחת.
org.tensorflow.Operand
פלט מופשט <T>
asOutput ()
מחזירה את הידית הסמלית של הטנזור.
תקציר T
asTensor ()
מחזיר את הטנזור באופרנד זה.
צורה מופשטת
צורה ()
מחזירה את הצורה (הידועה אולי בחלקה) של הטנזור שאליו מתייחס Output של אופרנד זה.
מחלקה מופשטת<T>
סוג ()
מחזירה את סוג הטנזור של אופרנד זה
org.tensorflow.ndarray.Shaped
מופשט int
דרגה ()
צורה מופשטת
צורה ()
מופשט ארוך
גודל ()
מחשב ומחזיר את הגודל הכולל של מיכל זה, במספר ערכים.

קבועים

מחרוזת סופית סטטית ציבורית OP_NAME

השם של המבצע הזה, כפי שידוע על ידי מנוע הליבה של TensorFlow

ערך קבוע: "TridiagonalSolve"

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

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

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

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

public static TridiagonalSolve <T> create ( scope scope, Operand <T> אלכסוני, Operand <T> rhs, Options... אפשרויות)

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

פרמטרים
תְחוּם ההיקף הנוכחי
אלכסונים טנסור של צורה `[..., 3, M]` ש-2 המימדים הפנימיים ביותר שלו מייצגים את המטריצות התלת-אלכסוניות כאשר שלוש שורות הן על-אלכסון, אלכסוני ותת-אלכסון, לפי הסדר. מתעלמים מהאלמנט האחרון של האלכסון העל ומהמרכיב הראשון של תת-אלכסון.
rhs טנסור של צורה `[..., M, K]`, המייצג K צלעות ימין לכל צד שמאל.
אפשרויות נושא ערכי תכונות אופציונליות
החזרות
  • מופע חדש של TridiagonalSolve

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

טנסור הצורה `[..., M, K]` המכיל את הפתרונות

public static TridiagonalSolve.Options partalPivoting (Boolian partialPivoting)

פרמטרים
Pivoting חלקית האם להחיל ציר חלקי. סיבוב חלקי הופך את ההליך ליציב יותר, אך איטי יותר.