PyTorch torch.geqrf 函数
torch.geqrf 是 PyTorch 中用于计算矩阵 QR 分解的函数,返回 Q 和 R 矩阵。
函数定义
torch.geqrf(input)
返回:
- a - 正交矩阵 Q
- tau - 产生正交矩阵所需的因子
使用示例
实例
import torch
# 创建矩阵
A = torch.randn(3, 3)
# QR分解
a, tau = torch.geqrf(A)
print("a (Q):", a)
print("tau:", tau)
# 创建矩阵
A = torch.randn(3, 3)
# QR分解
a, tau = torch.geqrf(A)
print("a (Q):", a)
print("tau:", tau)

Pytorch torch 参考手册