[ torch 参数更多 ] torch.qr

torch.qr

torch.qr(input, some=True, *, out=None)

paddle.linalg.qr

paddle.linalg.qr(x, mode='reduced', name=None)

PyTorch 相比 Paddle 支持更多其他参数,具体如下:

参数映射

PyTorch PaddlePaddle 备注
input x 表示输入 Tensor,仅参数名不一致。
some mode 表示 QR 分解的行为。 需进行转写。
out - 表示输出的 Tensor 元组。 Paddle 无此参数,需要转写。

转写示例

some:控制 QR 分解的行为

# 当进行完整的 QR 分解时
# PyTorch 写法
q, r = torch.qr(x, some=False)

# Paddle 写法
q, r = paddle.linalg.qr(x, mode='complete')

#当进行减少的 QR 分解时
# PyTorch 写法
q, r = torch.qr(x, some=True)

# Paddle 写法
q, r = paddle.linalg.qr(x, mode='reduced')

out:指定输出

# PyTorch 写法
torch.qr(x, out = (q, r) )

# Paddle 写法
q, r = paddle.linalg.qr(x)