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.3/api_docs/python/tf/raw_ops/TridiagonalSolve