[ 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)