TridiagonalSolve

الطبقة النهائية العامة TridiagonalSolve

يحل أنظمة المعادلات المثلثية.

يحل أنظمة المعادلات المثلثية. يدعم أبعاد الدفعة والجوانب اليمنى المتعددة لكل جانب أيسر. على وحدة المعالجة المركزية، يتم حساب الحل عن طريق الحذف Gaussian مع أو بدون التمحور الجزئي، اعتمادًا على سمة `partial_pivoting`. على وحدة معالجة الرسومات، يتم استخدام مكتبة cuSPARSE الخاصة بـ Nvidia: https://docs.nvidia.com/cuda/cusparse/index.html#gtsv لا يتم دعم التمحور الجزئي بعد بواسطة واجهات XLA الخلفية.

فئات متداخلة

فصل TridiagonalSolve.Options السمات الاختيارية لـ TridiagonalSolve

الثوابت

خيط OP_NAME اسم هذه العملية كما هو معروف بواسطة محرك TensorFlow الأساسي

الأساليب العامة

الإخراج <T>
كإخراج ()
إرجاع المقبض الرمزي للموتر.
ثابت <T يمتد TType > TridiagonalSolve <T>
إنشاء (نطاق النطاق ، المعامل <T> الأقطار، المعامل <T> rhs، الخيارات... الخيارات)
طريقة المصنع لإنشاء فئة تغلف عملية TridiagonalSolve جديدة.
الإخراج <T>
انتاج ()
موتر الشكل `[..., M, K]` الذي يحتوي على الحلول
TridiagonalSolve.Options ثابت
التمحور الجزئي (التمحور الجزئي المنطقي)

الطرق الموروثة

الثوابت

السلسلة النهائية الثابتة العامة OP_NAME

اسم هذه العملية كما هو معروف بواسطة محرك TensorFlow الأساسي

القيمة الثابتة: "TridiagonalSolve"

الأساليب العامة

الإخراج العام <T> كإخراج ()

إرجاع المقبض الرمزي للموتر.

المدخلات إلى عمليات TensorFlow هي مخرجات عملية TensorFlow أخرى. يتم استخدام هذه الطريقة للحصول على مقبض رمزي يمثل حساب الإدخال.

إنشاء TridiagonalSolve <T> ثابت عام ( نطاق النطاق ، المعامل <T> الأقطار، المعامل <T> rhs، الخيارات... خيارات)

طريقة المصنع لإنشاء فئة تغلف عملية TridiagonalSolve جديدة.

حدود
نِطَاق النطاق الحالي
الأقطار موتر الشكل `[..., 3, M]` الذي يمثل البعدان الأعمق له المصفوفات ثلاثية الأقطار بثلاثة صفوف هي الأقطار الفائقة والأقطار والأقطار الفرعية بالترتيب. يتم تجاهل العنصر الأخير من القطر الفائق والعنصر الأول من القطر الفرعي.
rhs موتر الشكل `[..., M, K]`، يمثل الجوانب K اليمنى لكل جانب أيسر.
خيارات يحمل قيم السمات الاختيارية
عائدات
  • مثيل جديد لـ TridiagonalSolve

الإخراج العام <T> الإخراج ()

موتر الشكل `[..., M, K]` الذي يحتوي على الحلول

TridiagonalSolve الثابت العام. الخيارات الجزئية (المحورية الجزئية المنطقية)

حدود
Pivoting ما إذا كان سيتم تطبيق التمحور الجزئي. المحوري الجزئي يجعل الإجراء أكثر استقرارًا، ولكنه أبطأ.