torch.linalg.ldl_solve¶
- torch.linalg.ldl_solve(LD, pivots, B, *, hermitian=False, out=None) Tensor ¶
使用 LDL 分解计算线性方程组的解。
LD
和pivots
是 LDL 分解的紧凑表示形式,预计由torch.linalg.ldl_factor_ex()
计算。此函数的hermitian
参数应与torch.linalg.ldl_factor_ex()
中的相应参数相同。支持 float、double、cfloat 和 cdouble 数据类型输入。还支持矩阵批处理,如果
A
是矩阵批处理,则输出具有相同的批处理维度。警告
此函数为“实验性”,它可能在未来的 PyTorch 版本中发生更改。
- 参数
- 关键字参数
示例
>>> A = torch.randn(2, 3, 3) >>> A = A @ A.mT # make symmetric >>> LD, pivots, info = torch.linalg.ldl_factor_ex(A) >>> B = torch.randn(2, 3, 4) >>> X = torch.linalg.ldl_solve(LD, pivots, B) >>> torch.linalg.norm(A @ X - B) >>> tensor(0.0001)