本文简要介绍python语言中 torch.cholesky
的用法。
用法:
torch.cholesky(input, upper=False, *, out=None) → Tensor
out(Tensor,可选的) -输出矩阵
计算对称正定矩阵 或对称正定矩阵批次的 Cholesky 分解。
如果
upper
是True
,则返回的矩阵U
是上三角矩阵,并且分解具有以下形式:如果
upper
是False
,则返回的矩阵L
是下三角矩阵,并且分解具有以下形式:如果
upper
是True
,并且 是一批对称正定矩阵,则返回的张量将由每个单独矩阵的上三角 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).transpose(-2, -1).conj()
此变换将为所有有效(对称正定)输入产生等效结果。
例子:
>>> a = torch.randn(3, 3) >>> a = torch.mm(a, a.t()) # make symmetric positive-definite >>> 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]]) >>> torch.mm(l, l.t()) 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 = torch.matmul(a, a.transpose(-1, -2)) + 1e-03 # make symmetric positive-definite >>> l = torch.cholesky(a) >>> z = torch.matmul(l, l.transpose(-1, -2)) >>> torch.max(torch.abs(z - a)) # Max non-zero tensor(2.3842e-07)
参数:
关键字参数:
相关用法
- Python PyTorch cholesky_ex用法及代码示例
- Python PyTorch cholesky_solve用法及代码示例
- Python PyTorch cholesky_inverse用法及代码示例
- Python PyTorch chunk用法及代码示例
- Python PyTorch chain_matmul用法及代码示例
- Python PyTorch checkpoint_sequential用法及代码示例
- Python PyTorch column_stack用法及代码示例
- Python PyTorch cumprod用法及代码示例
- Python PyTorch calculate_gain用法及代码示例
- Python PyTorch cov用法及代码示例
- Python PyTorch cos用法及代码示例
- Python PyTorch compute_deltas用法及代码示例
- Python PyTorch conv_transpose3d用法及代码示例
- Python PyTorch combinations用法及代码示例
- Python PyTorch conv2d用法及代码示例
- Python PyTorch cummax用法及代码示例
- Python PyTorch custom_from_mask用法及代码示例
- Python PyTorch collect_all用法及代码示例
- Python PyTorch convert用法及代码示例
- Python PyTorch conv1d用法及代码示例
- Python PyTorch cat用法及代码示例
- Python PyTorch constant_用法及代码示例
- Python PyTorch context用法及代码示例
- Python PyTorch count_nonzero用法及代码示例
- Python PyTorch cdist用法及代码示例
注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.cholesky。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。