Функция PyTorch torch.ormqr
Справочное руководство Pytorch torch
torch.ormqrЭто функция в PyTorch, использующая QR-разложение для умножения матриц. Она вычисляет Q @ input или input @ Q, где Q — ортогональная матрица, полученная из QR-разложения.
Определение функции
torch.ormqr(input, tau, left=True, transpose=False, out=None)
Parameters:
input(Tensor): Входная матрица.tau(Tensor): Коэффициенты отражателей Хаусхолдера.left(bool, необязательный): Если True, вычисляет Q @ input; иначе вычисляет input @ Q. По умолчанию True.transpose(bool, необязательный): Следует ли транспонировать Q. По умолчанию False.out(Tensor, необязательный): Выходной тензор.
Возвращаемое значение:
torch.Tensor: Возвращает результат матричного умножения.
Пример использования
Пример
import torch
# Создать матрицу
A = torch.randn(3, 3)
B = torch.randn(3, 4)
# QR-разложение
Q, R = torch.linalg.qr(A)
# Вычислить Q @ B
result = torch.ormqr(B, torch.zeros(3))
print("Форма матрицы B:", B.shape)
print("Форма результата:", result.shape)
# Создать матрицу
A = torch.randn(3, 3)
B = torch.randn(3, 4)
# QR-разложение
Q, R = torch.linalg.qr(A)
# Вычислить Q @ B
result = torch.ormqr(B, torch.zeros(3))
print("Форма матрицы B:", B.shape)
print("Форма результата:", result.shape)
Результат вывода:
矩阵 B 形状: torch.Size([3, 4]) 结果形状: torch.Size([3, 4])
Другие расширения