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's cuSPARSE: 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/versions/r2.9/api_docs/python/tf/raw_ops/TridiagonalSolve