torch.linalg.lu_factor_ex

torch.linalg.lu_factor_ex(A, *, pivot=True, check_errors=False, out=None) [исходный код]

Это версия lu_factor(), которая не выполняет проверку ошибок, если только check_errors= True. Она также возвращает тензор info, возвращаемый функцией LAPACK getrf.

Примечание

Если входные данные находятся на устройстве CUDA, эта функция синхронизируется только при условии, что check_errors= True.

Предупреждение

Эта функция является «экспериментальной» и может измениться в будущей версии PyTorch.

Параметры:

A (Tensor) – тензор формы (*, m, n), где * — ноль или более пакетных размерностей.

Именованные аргументы:

• pivot (bool, необязательно) – определяет, следует ли вычислять LU-разложение с частичным выбором главного элемента или обычное LU-разложение. pivot= False не поддерживается на CPU. Значение по умолчанию: True.
• check_errors (bool, необязательно) – определяет, проверять ли содержимое infos и вызывать ошибку, если оно не равно нулю. Значение по умолчанию: False.
• out (tuple, необязательно) – кортеж из трех тензоров, в которые записывается результат. Игнорируется, если None. Значение по умолчанию: None.

Возвращает:

Именованный кортеж (LU, pivots, info).