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).