tf.linalg.cholesky_solve

Решает системы линейных уравнений A X = RHS, используя факторизации Холеского.

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

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

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

tf.compat.v1.cholesky_solve, tf.compat.v1.linalg.cholesky_solve

tf.linalg.cholesky_solve(
    chol, rhs, name=None
)

Используется в ноутбуках

Используется в учебниках
Исследование TensorFlow Probability: Оценка ковариации

В частности, возвращает X из A X = RHS, где A = L L^T, L - это аргумент chol, а RHS - аргумент rhs.

# Solve 10 separate 2x2 linear systems:
A = ... # shape 10 x 2 x 2
RHS = ... # shape 10 x 2 x 1
chol = tf.linalg.cholesky(A)  # shape 10 x 2 x 2
X = tf.linalg.cholesky_solve(chol, RHS)  # shape 10 x 2 x 1
# tf.matmul(A, X) ~ RHS
X[3, :, 0]  # Solution to the linear system A[3, :, :] x = RHS[3, :, 0]

# Solve five linear systems (K = 5) for every member of the length 10 batch.
A = ... # shape 10 x 2 x 2
RHS = ... # shape 10 x 2 x 5
...
X[3, :, 2]  # Solution to the linear system A[3, :, :] x = RHS[3, :, 2]

Аргументы
`chol`	A `Tensor`. Должен быть `float32` или `float64`, форма имеет вид `[..., M, M]`. Факторизация Холеского `A`, например, `chol = tf.linalg.cholesky(A)`. По этой причине используются только нижние треугольные части (включая диагональ) последних двух измерений `chol`. Строго верхняя часть предполагается нулевой и не используется.
`rhs`	A `Tensor`, того же типа, что и `chol`, форма имеет вид `[..., M, K]`.
`name`	Имя для этой `Op`. По умолчанию - `cholesky_solve`.

Возвращаемые значения
Решение системы `A x = rhs`, форма `[..., M, K]`.

© 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/cholesky_solve