tf.linalg.triangular_solve

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

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

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

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

tf.compat.v1.linalg.triangular_solve, tf.compat.v1.matrix_triangular_solve

tf.linalg.triangular_solve(
    matrix, rhs, lower=True, adjoint=False, name=None
)

matrix — это тензор формы [..., M, M], чьи внутренние 2 измерения образуют квадратные матрицы. Если lower равно True, то строго верхняя треугольная часть каждой внутренней матрицы предполагается нулевой и не используется. Если lower равно False, то строго нижняя треугольная часть каждой внутренней матрицы предполагается нулевой и не используется. rhs — это тензор формы [..., M, N].

Выходной тензор имеет форму [..., M, N]. Если adjoint равно True, то внутренние матрицы в выходе удовлетворяют матричным уравнениям sum_k matrix[..., i, k] * output[..., k, j] = rhs[..., i, j]. Если adjoint равно False, то внутренние матрицы в выходе удовлетворяют матричным уравнениям sum_k adjoint(matrix[..., i, k]) * output[..., k, j] = rhs[..., i, j].

Пример:

a = tf.constant([[3,  0,  0,  0],
  [2,  1,  0,  0],
  [1,  0,  1,  0],
  [1,  1,  1,  1]], dtype=tf.float32)

b = tf.constant([[4], [2], [4], [2]], dtype=tf.float32)
x = tf.linalg.triangular_solve(a, b, lower=True)
x
<tf.Tensor: shape=(4, 1), dtype=float32, numpy=
array([[ 1.3333334 ],
       [-0.66666675],
       [ 2.6666665 ],
       [-1.3333331 ]], dtype=float32)>
tf.matmul(a, x)
<tf.Tensor: shape=(4, 1), dtype=float32, numpy=
array([[4.],
       [2.],
       [4.],
       [2.]], dtype=float32)>

Аргументы
`matrix`	A `Tensor`. Должен быть одного из следующих типов: `float64`, `float32`, `half`, `complex64`, `complex128`. Форма — `[..., M, M]`.
`rhs`	A `Tensor`. Должен иметь тот же тип, что и `matrix`. Форма — `[..., M, N]`.
`lower`	Необязательный `bool`. По умолчанию — `True`. Логическое значение, указывающее, являются ли внутренние матрицы в матрице нижними или верхними треугольными.
`adjoint`	Необязательный `bool`. По умолчанию — `False`. Логическое значение, указывающее, следует ли решать с матрицей или ее (блочной) сопряжённой.
`name`	Имя операции (необязательно).

Возвращаемое значение
A `Tensor`. Имеет тот же тип, что и матрица, и форму `[..., M, N]`.

© 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/linalg/triangular_solve