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, pivots, info) 的元组。