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’s getrf.

Примечание

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

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

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

Параметры:

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

Ключевые аргументы:

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

Возвращает:

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