[ torch 参数更多 ] torch.svd

torch.svd

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

paddle.linalg.svd

paddle.linalg.svd(x, full_matrics=False, name=None)

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

参数映射

PyTorch PaddlePaddle 备注
input x 输入 Tensor ,仅参数名不一致。
some full_matrics 表示需计算的奇异值数目。 与 PyTorch 默认值不同,需要转写。
compute_uv - 表示是否计算 U 和 V 。Paddle 无此参数,暂无转写方式。
out - 表示输出的 Tensor 元组。 Paddle 无此参数,需要转写。

转写示例

some:表示需计算的奇异值数目

# PyTorch 写法
u, s, v = torch.svd(x, some = True )

# Paddle 写法
u, s, v = paddle.linalg.svd(x, full_matrics = False)

out:指定输出

# PyTorch 写法
torch.svd(x, out=(u, s, v) )

# Paddle 写法
u, s, v = paddle.linalg.svd(x)