Python PyTorch cholesky_inverse用法及代碼示例

本文簡要介紹python語言中 torch.cholesky_inverse 的用法。

用法:

torch.cholesky_inverse(input, upper=False, *, out=None) → Tensor

參數：

• input(Tensor) -輸入二維張量 u ，一個上三角或下三角 Cholesky 因子

• upper(bool,可選的) -是否返回下(默認)或上三角矩陣

關鍵字參數：

out(Tensor,可選的) -inv 的輸出張量

使用 Cholesky 因子 u 計算對稱正定矩陣 A 的逆矩陣：返回矩陣 inv 。使用 LAPACK 例程 dpotri 和 spotri(以及相應的 MAGMA 例程)計算逆。

如果 upper 是 False ，則 u 是下三角形，因此返回的張量是

inv = (uu^{{T}})^{{-1}}

如果 upper 是 True 或未提供，則 u 是上三角形，因此返回的張量是

inv = (u^T u)^{{-1}}

例子：

>>> a = torch.randn(3, 3)
>>> a = torch.mm(a, a.t()) + 1e-05 * torch.eye(3) # make symmetric positive definite
>>> u = torch.cholesky(a)
>>> a
tensor([[  0.9935,  -0.6353,   1.5806],
        [ -0.6353,   0.8769,  -1.7183],
        [  1.5806,  -1.7183,  10.6618]])
>>> torch.cholesky_inverse(u)
tensor([[ 1.9314,  1.2251, -0.0889],
        [ 1.2251,  2.4439,  0.2122],
        [-0.0889,  0.2122,  0.1412]])
>>> a.inverse()
tensor([[ 1.9314,  1.2251, -0.0889],
        [ 1.2251,  2.4439,  0.2122],
        [-0.0889,  0.2122,  0.1412]])