tf.linalg.lu
Вычисляет LU-разложение одной или нескольких квадратных матриц.
tf.linalg.lu(
input: Annotated[Any, TV_Lu_T],
output_idx_type: TV_Lu_output_idx_type = tf.dtypes.int32,
name=None
)
Входные данные — это тензор формы [..., M, M], два самых внутренних измерения которого образуют квадратные матрицы.
Входные данные должны быть обратимыми.
Результат состоит из двух тензоров LU и P, содержащих LU-разложение всех входных подматриц [..., :, :]. LU кодирует нижнетреугольный и верхнетреугольный множители.
Для каждой входной подматрицы формы [M, M] матрица L является нижнетреугольной матрицей формы [M, M] с единичной диагональю, элементы которой соответствуют строго нижнетреугольной части LU. Матрица U является верхнетреугольной матрицей формы [M, M], элементы которой соответствуют верхнетреугольной части LU, включая диагональ.
P представляет матрицу перестановки, закодированную в виде списка индексов от 0 до M-1 включительно. Если P_mat обозначает матрицу перестановки, соответствующую P, то L, U и P удовлетворяют равенству P_mat * input = L * U.
| Аргументы | |
|---|---|
input | Тензор Tensor. Должен иметь один из следующих типов: float64, float32, half, complex64, complex128. Тензор формы [..., M, M], два самых внутренних измерения которого образуют матрицы размера [M, M]. |
output_idx_type | Необязательный тип tf.DType из списка: tf.int32, tf.int64. По умолчанию используется tf.int32. |
name | Имя операции (необязательно). |
| Возвращает | |
|---|---|
Кортеж объектов Tensor (lu, p). | |
lu | Тензор Tensor. Имеет тот же тип, что и input. |
p | Тензор Tensor типа output_idx_type. |
© 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/lu