Функция linalg.solve
Функция linalg.solve решает систему линейных уравнений
вида AX = B относительно неизвестной матрицы X.
Первым параметром передается матрица коэффициентов A,
вторым - матрица или вектор правых частей B.
Функция возвращает тензор X, который является решением системы.
Матрица A должна быть квадратной и невырожденной.
Синтаксис
tf.linalg.solve(A, B, [name])
Пример
Давайте решим систему линейных уравнений с матрицей коэффициентов
[[2, 1], [1, 3]] и вектором правых частей [5, 10]:
import tensorflow as tf
A = tf.constant([[2, 1], [1, 3]], dtype=tf.float32)
B = tf.constant([5, 10], dtype=tf.float32)
res = tf.linalg.solve(A, B)
print(res)
Результат выполнения кода:
tf.Tensor([1. 3.], shape=(2,), dtype=float32)
Пример
Давайте решим систему с матрицей правых частей, содержащей несколько столбцов:
import tensorflow as tf
A = tf.constant([[2, 1], [1, 3]], dtype=tf.float32)
B = tf.constant([[5, 1], [10, 2]], dtype=tf.float32)
res = tf.linalg.solve(A, B)
print(res)
Результат выполнения кода:
tf.Tensor(
[[1. 0.2]
[3. 0.6]], shape=(2, 2), dtype=float32)
Пример
Давайте решим систему с единичной матрицей коэффициентов:
import tensorflow as tf
A = tf.constant([[1, 0], [0, 1]], dtype=tf.float32)
B = tf.constant([4, 7], dtype=tf.float32)
res = tf.linalg.solve(A, B)
print(res)
Результат выполнения кода:
tf.Tensor([4. 7.], shape=(2,), dtype=float32)
