[torch 参数更多]torch.linalg.eigh¶

torch.linalg.eigh ¶

          torch.linalg.eigh(A, UPLO='L', *, out=None)

         

          paddle.linalg.eigh(x, UPLO='L', name=None)

         

其中 PyTorch 相比 Paddle 支持更多其他参数，具体如下：

PyTorch	PaddlePaddle	备注
A	x	输入 Tensor，仅参数名不一致。
UPLO	UPLO	表示计算上三角或者下三角矩阵
out	-	表示输出的 Tensor ， Paddle 无此参数，需要进行转写。

           # PyTorch 写法:
torch.linalg.eigh(x, out=y)

# Paddle 写法:
paddle.assign(paddle.linalg.eigh(x), y)