يحل أنظمة المعادلات المثلثية.
يحل أنظمة المعادلات المثلثية. يدعم أبعاد الدفعة والجوانب اليمنى المتعددة لكل جانب أيسر. على وحدة المعالجة المركزية، يتم حساب الحل عن طريق الحذف 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 الأساسي
الأساليب العامة
الإخراج العام <T> كإخراج ()
إرجاع المقبض الرمزي للموتر.
المدخلات إلى عمليات TensorFlow هي مخرجات عملية TensorFlow أخرى. يتم استخدام هذه الطريقة للحصول على مقبض رمزي يمثل حساب الإدخال.
إنشاء TridiagonalSolve <T> ثابت عام ( نطاق النطاق ، المعامل <T> الأقطار، المعامل <T> rhs، الخيارات... خيارات)
طريقة المصنع لإنشاء فئة تغلف عملية TridiagonalSolve جديدة.
حدود
نِطَاق | النطاق الحالي |
---|---|
الأقطار | موتر الشكل `[..., 3, M]` الذي يمثل البعدان الأعمق له المصفوفات ثلاثية الأقطار بثلاثة صفوف هي الأقطار الفائقة والأقطار والأقطار الفرعية بالترتيب. يتم تجاهل العنصر الأخير من القطر الفائق والعنصر الأول من القطر الفرعي. |
rhs | موتر الشكل `[..., M, K]`، يمثل الجوانب K اليمنى لكل جانب أيسر. |
خيارات | يحمل قيم السمات الاختيارية |
عائدات
- مثيل جديد لـ TridiagonalSolve
TridiagonalSolve الثابت العام. الخيارات الجزئية (المحورية الجزئية المنطقية)
حدود
Pivoting | ما إذا كان سيتم تطبيق التمحور الجزئي. المحوري الجزئي يجعل الإجراء أكثر استقرارًا، ولكنه أبطأ. |
---|