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

org.tensorflow.op.RawOp
końcowa wartość logiczna
równa się (obiekt obiektu)
końcowy wew
hashCode ()
Operacja
op ()
Zwróć tę jednostkę obliczeniową jako pojedynczą Operation .
ostatni ciąg
doString ()
wartość logiczna
równa się (Obiekt arg0)
ostatnia klasa<?>
pobierzKlasę ()
wew
hashCode ()
ostateczna pustka
powiadomić ()
ostateczna pustka
powiadom wszystkich ()
Strunowy
doString ()
ostateczna pustka
czekaj (długi arg0, int arg1)
ostateczna pustka
czekaj (długi arg0)
ostateczna pustka
Czekać ()
org.tensorflow.op.Op
abstrakcyjne środowisko wykonania
środowisko ()
Zwróć środowisko wykonawcze, w którym utworzono tę operację.
abstrakcyjna operacja
op ()
Zwróć tę jednostkę obliczeniową jako pojedynczą Operation .
org.tensorflow.Operand
streszczenie Wyjście <T>
jako wyjście ()
Zwraca symboliczny uchwyt tensora.
streszczenie T
asTensor ()
Zwraca tensor tego operandu.
abstrakcyjny kształt
kształt ()
Zwraca (prawdopodobnie częściowo znany) kształt tensora, do którego odnosi się dane Output tego operandu.
Klasa abstrakcyjna<T>
typ ()
Zwraca typ tensora tego operandu
org.tensorflow.ndarray.Shaped
streszczenie wew
ranga ()
abstrakcyjny kształt
kształt ()
abstrakcyjne, długie
rozmiar ()
Oblicza i zwraca całkowity rozmiar tego kontenera w liczbie wartości.

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.