快捷方式

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) 的张量,其中 * 是零个或多个批次维度。

关键字参数
  • left (布尔值, 可选) – 是否求解系统 AX=BAX=BXA=BXA = B。默认值:True

  • check_errors (布尔值, 可选) – 控制是否检查 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)

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源