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)

שיטות בירושה

קבועים

מחרוזת סופית סטטית ציבורית 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 חלקית האם להחיל ציר חלקי. סיבוב חלקי הופך את ההליך ליציב יותר, אך איטי יותר.