שיעור הגמר הציבורי TridiagonalSolve
פותר מערכות משוואות תלת אלכסוניות.
פותר מערכות משוואות תלת אלכסוניות. תומך במידות אצווה ובמספר צדדים ימניים לכל צד שמאל. ב-CPU, הפתרון מחושב באמצעות חיסול גאוסי עם או בלי סיבוב חלקי, בהתאם לתכונה 'חלקית_ציר'. ב-GPU, נעשה שימוש בספריית cuSPARSE של Nvidia: https://docs.nvidia.com/cuda/cusparse/index.html#gtsv Pivoting חלקי עדיין לא נתמך על ידי קצה אחורי של XLA.
כיתות מקוננות
| מעמד | TridiagonalSolve.Options | תכונות אופציונליות עבור TridiagonalSolve | |
שיטות ציבוריות
| פלט <T> | asOutput () מחזירה את הידית הסמלית של טנזור. |
| סטטי <T> TridiagonalSolve <T> | ליצור ( היקף היקף , אלכסוני Operand <T>, Operand <T> rhs, אפשרויות... אפשרויות) שיטת מפעל ליצירת מחלקה העוטפת פעולת TridiagonalSolve חדשה. |
| פלט <T> | פלט () טנסור הצורה `[..., M, K]` המכיל את הפתרונות |
| סטטי TridiagonalSolve.Options | partialPivoting (בוליאני partialPivoting) |
שיטות בירושה
שיטות ציבוריות
פלט ציבורי <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
public static TridiagonalSolve.Options partalPivoting (Boolian partialPivoting)
פרמטרים
| Pivoting חלקית | האם להחיל ציר חלקי. סיבוב חלקי הופך את ההליך ליציב יותר, אך איטי יותר. |
|---|