torch.linalg.lu_factor_ex

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

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

注意

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

警告

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

Parameters

A (张量) – 形状为 (*, m, n) 的张量，其中 * 表示零个或多个批次维度。

Keyword Arguments

• pivot (bool, 可选) – 是否使用部分旋转计算LU分解，或使用常规LU分解。pivot= False 在CPU上不支持。默认值：True。

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

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

Returns

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