[torch 参数更多]torch.Tensor.symeig¶
torch.Tensor.symeig¶
# pytorch1.9 以上版本不支持
torch.Tensor.symeig(eigenvectors=False, upper=True)
paddle.linalg.eigh¶
# eigenvectors 为 True
paddle.linalg.eigh(x, UPLO='L', name=None)
# eigenvectors 为 False
paddle.linalg.eigvalsh(x, UPLO='L', name=None)
Pytorch 相比 Paddle 支持更多其他参数，具体如下：
参数映射¶
| PyTorch | PaddlePaddle | 备注 |
|---|---|---|
| - | x | 表示输入的 Tensor 。 |
| eigenvectors | - | 表示是否计算 eigenvectors，Paddle 无此参数，需要转写。 |
| upper | UPLO | 表示计算上三角或者下三角矩阵，PyTorch 取值 bool 类型，Paddle 取值 L, U。 |
转写示例¶
eigenvectors：表示是否计算特征向量¶
# Pytorch 写法，eigenvectors 为 False
e, _ = x.symeig(eigenvectors=False)
# Paddle 写法
e = paddle.linalg.eigvalsh(x)
# Pytorch 写法，eigenvectors 为 True
e, v = x.symeig(eigenvectors=True)
# Paddle 写法
e, v = paddle.linalg.eigh(x)
upper：表示计算上三角或者下三角矩阵¶
# Pytorch 写法
e, v = x.symeig(upper = False)
# Paddle 写法
e, v = paddle.linalg.eigh(x, UPLO = 'L')