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 используется библиотека 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`. Применять ли частичный поворот. Частичный поворот делает процедуру более устойчивой, но медленнее.
`perturb_singular`	Необязательный `bool`. По умолчанию `False`.
`name`	Название операции (необязательно).

Возвращаемое значение
A `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