当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python PyTorch cholesky_inverse用法及代码示例


本文简要介绍python语言中 torch.cholesky_inverse 的用法。

用法:

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

参数

  • input(Tensor) -输入二维张量 ,一个上三角或下三角 Cholesky 因子

  • upper(bool,可选的) -是否返回下(默认)或上三角矩阵

关键字参数

out(Tensor,可选的) -inv 的输出张量

使用 Cholesky 因子 计算对称正定矩阵 的逆矩阵:返回矩阵 inv 。使用 LAPACK 例程 dpotrispotri(以及相应的 MAGMA 例程)计算逆。

如果 upperFalse ,则 是下三角形,因此返回的张量是

如果 upperTrue 或未提供,则 是上三角形,因此返回的张量是

例子:

>>> 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]])

相关用法


注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.cholesky_inverse。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。