TridiagonalSolve

publiczna klasa końcowa TridiagonalSolve

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>
utwórz (Zakres zasięgu , Przekątne argumentu <T>, Argument <T> po prawej stronie, Opcje... opcje)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację TridiagonalSolve.
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

Wartość stała: „TridiagonalSolve”

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

publiczne wyjście <T> wyjście ()

Tensor kształtu „[..., M, K]” zawierający rozwiązania

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.