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 (Tensor) – 形状为 (*, n, n) 的张量，其中 * 是零个或多个批次维度。

关键字参数

• left (bool, optional) – 是否求解系统 $AX=B$ 或 $XA = B$。默认值：True。

• check_errors (bool, optional) – 控制是否检查 infos 的内容，并在其为非零时引发错误。默认值：False。

• out (tuple, optional) – 用于写入输出的两个张量的元组。如果为 None，则忽略。默认值：None。

返回值

一个名为 (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)