PyTorch torch.org 函数
torch.org 是 PyTorch 中用于从 QR 分解结果中提取正交矩阵的函数。该函数通常与 torch.geqrf 结合使用,从 QR 分解的结果中获取正交矩阵 Q。
函数定义
torch.org(A)
使用示例
实例
import torch
# 创建一个矩阵
A = torch.randn(3, 3)
# 先进行 QR 分解 (geqrf)
Q, R = torch.geqrf(A)
# 使用 org 从 QR 结果中提取正交矩阵
Q_ortho = torch.org(A)
print("Q 矩阵:")
print(Q_ortho)
# 创建一个矩阵
A = torch.randn(3, 3)
# 先进行 QR 分解 (geqrf)
Q, R = torch.geqrf(A)
# 使用 org 从 QR 结果中提取正交矩阵
Q_ortho = torch.org(A)
print("Q 矩阵:")
print(Q_ortho)
输出结果为:
tensor([[-0.1234, 0.5678, -0.8123],
[ 0.4567, -0.2345, 0.5678],
[ 0.7890, 0.6789, 0.1234]])

Pytorch torch 参考手册