快捷方式

torch.linalg.lu_factor

torch.linalg.lu_factor(A, *, bool pivot=True, out=None) -> (Tensor, Tensor)

计算矩阵进行部分主元 LU 分解的紧凑表示。

此函数计算由 torch.linalg.lu() 给出的分解的紧凑表示。如果矩阵是方阵,则此表示可用于 torch.linalg.lu_solve() 来解决共享矩阵 A 的线性方程组。

返回的分解表示为命名元组 (LU, pivots)LU 矩阵与输入矩阵 A 的形状相同。其上三角部分和下三角部分分别编码 A 的 LU 分解的 LU 的非常数元素。

返回的置换矩阵由一个 1 索引向量表示。 pivots[i] == j 表示在算法的第 i 步中,第 i 行与第 j-1 行进行了置换。

在 CUDA 上,可以使用 pivot= False。在这种情况下,如果存在,则此函数返回不进行主元的 LU 分解。

支持 float、double、cfloat 和 cdouble 数据类型的输入。也支持矩阵批次,如果输入是矩阵批次,则输出具有相同的批次维度。

注意

当输入在 CUDA 设备上时,此函数会将该设备与 CPU 同步。有关不进行同步的此函数的版本,请参阅 torch.linalg.lu_factor_ex()

警告

LU 分解几乎从不唯一,因为通常存在不同的置换矩阵可以产生不同的 LU 分解。因此,不同的平台(如 SciPy)或不同设备上的输入可能会产生不同的有效分解。

仅当输入矩阵是满秩时才支持梯度计算。如果不满足此条件,不会抛出错误,但梯度可能不是有限的。这是因为带主元的 LU 分解在这些点不可微。

另请参阅

torch.linalg.lu_solve() 求解线性方程组,前提是输入矩阵是方阵且可逆,并且给定此函数的输出。

torch.lu_unpack()lu_factor() 返回的张量解包成构成分解的三个矩阵 P, L, U

torch.linalg.lu() 计算可能非方阵的部分主元 LU 分解。它是 lu_factor()torch.lu_unpack() 的组合。

torch.linalg.solve() 求解线性方程组。它是 lu_factor()lu_solve() 的组合。

参数

A (张量) – 形状为 (*, m, n) 的张量,其中 * 是零个或多个批次维度。

关键字参数
  • pivot (布尔值, 可选) – 是否计算带部分主元的 LU 分解或常规 LU 分解。 pivot= False 在 CPU 上不受支持。默认值:True

  • out (元组, 可选) – 要将输出写入的两个张量的元组。如果为 None 则忽略。默认值:None

返回

命名元组 (LU, pivots)

引发

RuntimeError – 如果 A 矩阵不可逆或批处理 A 中的任何矩阵不可逆。

示例

>>> A = torch.randn(2, 3, 3)
>>> B1 = torch.randn(2, 3, 4)
>>> B2 = torch.randn(2, 3, 7)
>>> LU, pivots = torch.linalg.lu_factor(A)
>>> X1 = torch.linalg.lu_solve(LU, pivots, B1)
>>> X2 = torch.linalg.lu_solve(LU, pivots, B2)
>>> torch.allclose(A @ X1, B1)
True
>>> torch.allclose(A @ X2, B2)
True

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获取问题的答案

查看资源