现在位置: 首页 > PyTorch 教程 > 正文

PyTorch torch.geqrf 函数


Pytorch torch 参考手册 Pytorch torch 参考手册

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)

Pytorch torch 参考手册 Pytorch torch 参考手册