torch.linalg.lu_factor_ex

torch.linalg.lu_factor_ex(A, *, pivot=True, check_errors=False, out=None)

这是一个不执行错误检查的版本 lu_factor() ，除非 check_errors = True。它还返回 LAPACK 的 getrf 返回的 info 张量。

注意

当输入位于 CUDA 设备上时，此函数仅在 check_errors = True 时同步。

警告

此函数为“实验性”的，它可能在未来的 PyTorch 版本中发生变化。

参数:

A（张量）- 形状为(*, m, n)的张量，其中*表示零个或多个批处理维度。

关键字参数:

• pivot（布尔值，可选）- 是否使用部分主元置换来计算 LU 分解，或使用常规 LU 分解。 pivot = False 在 CPU 上不支持。默认：True。

• check_errors（布尔值，可选）- 控制是否检查 infos 的内容，如果非零则引发错误。默认：False。

• out（元组，可选）- 要写入的三个张量的元组。如果为 None 则忽略。默认：None。

返回值:

命名元组（LU，置换，信息）。