Rozwiązuje trójdiagonalne układy równań.
Rozwiązuje trójdiagonalne układy równań. Obsługuje wymiary partii i wiele prawych stron na każdą lewą stronę. Na procesorze rozwiązanie jest obliczane poprzez eliminację Gaussa z częściowym obracaniem lub bez, w zależności od atrybutu „partial_pivoting”. Na GPU używana jest biblioteka cuSPARSE firmy Nvidia: https://docs.nvidia.com/cuda/cusparse/index.html#gtsv Częściowe obracanie nie jest jeszcze obsługiwane przez backendy XLA.
Klasy zagnieżdżone
| klasa | Opcje rozwiązania trójdiagonalnego | Opcjonalne atrybuty TridiagonalSolve | |
Stałe
| Strunowy | OP_NAME | Nazwa tej operacji znana silnikowi rdzenia TensorFlow | 
Metody publiczne
| Wyjście <T> |  jako wyjście ()  Zwraca symboliczny uchwyt tensora. | 
| statyczny <T rozszerza TType > TridiagonalSolve <T> | |
| Wyjście <T> |  wyjście ()  Tensor kształtu „[..., M, K]” zawierający rozwiązania | 
| statyczne TridiagonalSolve.Options |  częściowePivoting (Boolean częściowePivoting)  | 
Metody dziedziczone
Stałe
publiczny statyczny końcowy ciąg znaków OP_NAME
Nazwa tej operacji znana silnikowi rdzenia TensorFlow
Metody publiczne
publiczne wyjście <T> asOutput ()
Zwraca symboliczny uchwyt tensora.
Dane wejściowe operacji TensorFlow są wynikami innej operacji TensorFlow. Ta metoda służy do uzyskania symbolicznego uchwytu reprezentującego obliczenia danych wejściowych.
public static TridiagonalSolve <T> create (zakres zakresu , argument <T> po przekątnej, argument <T> po prawej stronie, opcje... opcje)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację TridiagonalSolve.
Parametry
| zakres | aktualny zakres | 
|---|---|
| przekątne | Tensor kształtu „[..., 3, M]”, którego 2 najbardziej wewnętrzne wymiary reprezentują macierze trójdiagonalne, w których trzy rzędy to kolejno naddiagonalny, przekątny i subdiagonalny. Ostatni element nadprzekątnej i pierwszy element podprzekątnej są ignorowane. | 
| prawa strona | Tensor kształtu „[..., M, K]”, reprezentujący K prawych stron na każdą lewą stronę. | 
| opcje | przenosi opcjonalne wartości atrybutów | 
Zwroty
- nowa instancja TridiagonalSolve
public static TridiagonalSolve.Options częściowePivoting (Boolean częściowePivoting)
Parametry
| częścioweObracanie | Czy zastosować częściowe obracanie. Częściowe obracanie sprawia, że procedura jest stabilniejsza, ale wolniejsza. | 
|---|