torch.linalg.solve_ex¶
- torch.linalg.solve_ex(A, B, *, left=True, check_errors=False, out=None)¶
solve()
的一个版本,它不执行错误检查,除非check_errors
= True。它还会返回 LAPACK 的 getrf 返回的info
张量。注意
当输入位于 CUDA 设备上时,此函数仅在
check_errors
= True 时同步。警告
此函数为“实验性”,它可能在未来的 PyTorch 版本中发生变化。
- 参数
A (张量) – 形状为 (*, n, n) 的张量,其中 * 是零个或多个批次维度。
- 关键字参数
- 返回值
一个名为 (result, info) 的命名元组。
示例
>>> A = torch.randn(3, 3) >>> Ainv, info = torch.linalg.solve_ex(A) >>> torch.dist(torch.linalg.inv(A), Ainv) tensor(0.) >>> info tensor(0, dtype=torch.int32)