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. |
---|