tf.raw_ops.TridiagonalSolve

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

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

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

Для получения дополнительной информации см. Руководство по миграции.

tf.compat.v1.raw_ops.TridiagonalSolve

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

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

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

Возвращаемые значения
A `Tensor`. Имеет тот же тип, что и `diagonals`.

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