tf.raw_ops.TridiagonalSolve

Решает системы уравнений с трёхдиагональными матрицами.

Просмотреть псевдонимы

Псевдонимы совместимости для миграции

Подробнее см. в руководстве по миграции.

tf.compat.v1.raw_ops.TridiagonalSolve

tf.raw_ops.TridiagonalSolve(
    diagonals, rhs, partial_pivoting=True, perturb_singular=False, name=None
)

Решает системы уравнений с трёхдиагональными матрицами. Поддерживает пакетные измерения и несколько правых частей для каждой левой части. На CPU решение вычисляется с помощью метода Гаусса с частичным выбором главного элемента или без него — в зависимости от атрибута partial_pivoting. На GPU используется библиотека Nvidia cuSPARSE: https://docs.nvidia.com/cuda/cusparse/index.html#gtsv Частичный выбор главного элемента пока не поддерживается бэкендами XLA.

Аргументы
`diagonals`	Объект `Tensor`. Должен иметь один из следующих типов: `float64`, `float32`, `complex64`, `complex128`. Тензор формы `[..., 3, M]`, два внутренних измерения которого представляют трёхдиагональные матрицы: три строки соответствуют наддиагоналям, диагоналям и поддиагоналям именно в таком порядке. Последний элемент наддиагонали и первый элемент поддиагонали игнорируются.
`rhs`	Объект `Tensor`. Должен иметь тот же тип, что и `diagonals`. Тензор формы `[..., M, K]`, представляющий K правых частей для каждой левой части.
`partial_pivoting`	Необязательный объект `bool`. Значение по умолчанию — `True`. Определяет, следует ли применять частичный выбор главного элемента. Частичный выбор главного элемента повышает устойчивость процедуры, но замедляет её.
`perturb_singular`	Необязательный объект `bool`. Значение по умолчанию — `False`.
`name`	Имя операции (необязательно).

Возвращает
Объект `Tensor`. Имеет тот же тип, что и `diagonals`.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/raw_ops/TridiagonalSolve