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, optional) – 是否使用部分主元进行 LU 分解，或进行常规 LU 分解。pivot= False 在 CPU 上不受支持。默认为True。

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

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

返回

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