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

关键字参数

• pivot (bool, 可选) – 是否计算带有部分选主元的LU分解，或者常规的LU分解。在CPU上pivot = False 不被支持。默认值： True。

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

• out (元组, 可选) – 要写入输出的三个张量组成的元组。如果为None则忽略。默认值：None。

返回值

(LU, pivots, info) 这个命名元组。