torch.cholesky

torch.cholesky(input, upper=False, *, out=None) → 张量

计算对称正定矩阵 $A$ 的 Cholesky 分解，或对一批对称正定矩阵进行 Cholesky 分解。

如果 upper 是 True，则返回的矩阵 U 是上三角矩阵，并且分解的形式为：

$$A = U^TU$$

如果 upper 是 False，则返回的矩阵 L 是下三角矩阵，并且分解的形式为：

$$A = LL^T$$

如果 upper 是 True，并且 $A$ 是一批对称正定矩阵，那么返回的张量将由每个单独矩阵的上三角Cholesky因子组成。同样，当 upper 是 False 时，返回的张量将由每个单独矩阵的下三角Cholesky因子组成。

警告

torch.cholesky() 已被弃用，取而代之的是 torch.linalg.cholesky() 并将在未来的 PyTorch 版本中移除。

L = torch.cholesky(A) 应替换为

L = torch.linalg.cholesky(A)

U = torch.cholesky(A, upper=True) 应替换为

U = torch.linalg.cholesky(A).mH

此转换将对所有有效的（对称正定）输入产生等效结果。

Parameters

• 输入 (张量) – 输入张量 $A$ 的大小为 $(*, n, n)$ 其中 * 是零个或多个由对称正定矩阵组成的批次维度。

• upper (bool, 可选) – 标志，指示是否返回上三角矩阵或下三角矩阵。默认值：False

Keyword Arguments

输出 (张量, 可选) – 输出矩阵

示例：

>>> a = torch.randn(3, 3)
>>> a = a @ a.mT + 1e-3 # 使对称正定
>>> l = torch.cholesky(a)
>>> a
tensor([[ 2.4112, -0.7486,  1.4551],
        [-0.7486,  1.3544,  0.1294],
        [ 1.4551,  0.1294,  1.6724]])
>>> l
tensor([[ 1.5528,  0.0000,  0.0000],
        [-0.4821,  1.0592,  0.0000],
        [ 0.9371,  0.5487,  0.7023]])
>>> l @ l.mT
tensor([[ 2.4112, -0.7486,  1.4551],
        [-0.7486,  1.3544,  0.1294],
        [ 1.4551,  0.1294,  1.6724]])
>>> a = torch.randn(3, 2, 2) # 批量输入的示例
>>> a = a @ a.mT + 1e-03 # 使对称正定
>>> l = torch.cholesky(a)
>>> z = l @ l.mT
>>> torch.dist(z, a)
tensor(2.3842e-07)