समीकरणों की त्रिविकोणीय प्रणालियों को हल करता है।
समीकरणों की त्रिविकोणीय प्रणालियों को हल करता है। प्रत्येक बाएँ हाथ की ओर बैच आयामों और एकाधिक दाएँ पक्ष का समर्थन करता है। CPU पर, समाधान की गणना गाऊसी उन्मूलन के माध्यम से की जाती है, जिसमें आंशिक पिवोटिंग के साथ या बिना, `partial_pivoting` विशेषता के आधार पर गणना की जाती है। GPU पर, Nvidia की cuSPARSE लाइब्रेरी का उपयोग किया जाता है: https://docs.nvidia.com/cuda/cusparse/index.html#gtsv आंशिक पिवोटिंग अभी तक XLA बैकएंड द्वारा समर्थित नहीं है।
नेस्टेड कक्षाएं
कक्षा | TridiagonalSolve.Options | के लिए वैकल्पिक विशेषताओं TridiagonalSolve |
सार्वजनिक तरीके
आउटपुट <टी> | asOutput () टेंसर का प्रतीकात्मक हैंडल लौटाता है। |
स्थिर <टी> TridiagonalSolve <टी> | बनाने ( स्कोप गुंजाइश, ओपेरैंड <टी> विकर्ण, ओपेरैंड <टी> RHS, विकल्प ... विकल्प) एक नया TridiagonalSolve ऑपरेशन रैपिंग क्लास बनाने के लिए फ़ैक्टरी विधि। |
आउटपुट <टी> | उत्पादन () आकार का टेंसर [..., M, K] समाधान युक्त |
स्थिर TridiagonalSolve.Options | partialPivoting (बूलियन partialPivoting) |
स्थिर TridiagonalSolve.Options | perturbSingular (बूलियन perturbSingular) |
विरासत में मिली विधियां
सार्वजनिक तरीके
सार्वजनिक आउटपुट <टी> asOutput ()
टेंसर का प्रतीकात्मक हैंडल लौटाता है।
TensorFlow संचालन के लिए इनपुट एक अन्य TensorFlow ऑपरेशन के आउटपुट हैं। इस पद्धति का उपयोग एक प्रतीकात्मक हैंडल प्राप्त करने के लिए किया जाता है जो इनपुट की गणना का प्रतिनिधित्व करता है।
सार्वजनिक स्थिर TridiagonalSolve <टी> बनाने ( स्कोप गुंजाइश, ओपेरैंड <टी> विकर्ण, ओपेरैंड <टी> RHS, विकल्प ... विकल्प)
एक नया TridiagonalSolve ऑपरेशन रैपिंग क्लास बनाने के लिए फ़ैक्टरी विधि।
मापदंडों
दायरा | वर्तमान दायरा |
---|---|
विकर्णों | आकार का टेंसर `[..., 3, M] जिसका अंतरतम 2 आयाम त्रिविकोणीय मैट्रिक्स का प्रतिनिधित्व करते हैं, जिसमें तीन पंक्तियाँ सुपरडायगोनल, विकर्ण और उप-विकर्ण होती हैं। सुपरडायगोनल के अंतिम तत्व और सबडिगोनल के पहले तत्व को नजरअंदाज कर दिया जाता है। |
आरएचएस | आकार का टेंसर `[..., M, K]`, प्रत्येक बाएँ हाथ के प्रति K दाएँ पक्ष का प्रतिनिधित्व करता है। |
विकल्प | वैकल्पिक विशेषता मान रखता है |
रिटर्न
- TridiagonalSolve का एक नया उदाहरण
सार्वजनिक स्थिर TridiagonalSolve.Options partialPivoting (बूलियन partialPivoting)
मापदंडों
आंशिक पिवोटिंग | क्या आंशिक पिवोटिंग लागू करना है। आंशिक पिवोटिंग प्रक्रिया को अधिक स्थिर, लेकिन धीमा बनाता है। |
---|