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, 可选) – 是否求解方程组 $$AX=B$$ 或者 $$XA = B$$ 。默认值： True。
check_errors (bool, optional) – 控制是否检查 infos 的内容并在其非零时触发错误。默认值：False。
out (元组, 可选) – 用于写入输出的两个张量的元组。如果为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)